Google Git
Sign in
chromium / external / github.com / tensorflow / tensorflow / 22380724dd052476dc4afdefdcea00f17f880cd0 / . / tensorflow / core / kernels
tree: f49b7f9df2f2464a192706d7c0a003e6607e049a [path history] [tgz]
  1. batching_util/
  2. boosted_trees/
  3. data/
  4. fuzzing/
  5. hexagon/
  6. image/
  7. linalg/
  8. mkl/
  9. mlir_generated/
  10. neon/
  11. rnn/
  12. sparse/
  13. special_math/
  14. spectrogram_test_data/
  15. tensor_forest/
  16. aggregate_ops.cc
  17. aggregate_ops.h
  18. aggregate_ops_cpu.h
  19. aggregate_ops_gpu.cu.cc
  20. argmax_op.cc
  21. argmax_op.h
  22. argmax_op_gpu.cu.cc
  23. as_string_op.cc
  24. as_string_op_test.cc
  25. assign_op.h
  26. avgpooling_op.cc
  27. avgpooling_op.h
  28. avgpooling_op_gpu.cu.cc
  29. barrier_ops.cc
  30. base64_ops.cc
  31. basic_ops_benchmark_test.cc
  32. batch_kernels.cc
  33. batch_matmul_op_complex.cc
  34. batch_matmul_op_impl.h
  35. batch_matmul_op_real.cc
  36. batch_matmul_op_test.cc
  37. batch_norm_op.cc
  38. batch_norm_op.h
  39. batch_norm_op_gpu.cu.cc
  40. batch_norm_op_test.cc
  41. batch_util.h
  42. batchtospace_op.cc
  43. bcast_ops.cc
  44. betainc_op.cc
  45. betainc_op.h
  46. betainc_op_gpu.cu.cc
  47. bias_op.cc
  48. bias_op.h
  49. bias_op_gpu.cu.cc
  50. bias_op_gpu.h
  51. bias_op_test.cc
  52. bincount_op.cc
  53. bincount_op.h
  54. bincount_op_gpu.cu.cc
  55. bincount_op_test.cc
  56. broadcast_to_op.cc
  57. broadcast_to_op.h
  58. broadcast_to_op_gpu.cu.cc
  59. broadcast_to_op_test.cc
  60. bucketize_op.cc
  61. bucketize_op.h
  62. bucketize_op_gpu.cu.cc
  63. BUILD
  64. candidate_sampler_ops.cc
  65. cast_op.cc
  66. cast_op.h
  67. cast_op_gpu.cu.cc
  68. cast_op_impl.h
  69. cast_op_impl_bfloat.cc
  70. cast_op_impl_bool.cc
  71. cast_op_impl_complex128.cc
  72. cast_op_impl_complex64.cc
  73. cast_op_impl_double.cc
  74. cast_op_impl_float.cc
  75. cast_op_impl_half.cc
  76. cast_op_impl_int16.cc
  77. cast_op_impl_int32.cc
  78. cast_op_impl_int64.cc
  79. cast_op_impl_int8.cc
  80. cast_op_impl_uint16.cc
  81. cast_op_impl_uint32.cc
  82. cast_op_impl_uint64.cc
  83. cast_op_impl_uint8.cc
  84. cast_op_test.cc
  85. check_numerics_op.cc
  86. check_numerics_op_gpu.cu.cc
  87. clustering_ops.cc
  88. clustering_ops_test.cc
  89. collective_nccl.cc
  90. collective_nccl.h
  91. collective_nccl_broadcaster.cc
  92. collective_nccl_broadcaster.h
  93. collective_nccl_gatherer.cc
  94. collective_nccl_gatherer.h
  95. collective_nccl_reducer.cc
  96. collective_nccl_reducer.h
  97. collective_nccl_test.cc
  98. collective_ops.cc
  99. compare_and_bitpack_op.cc
  100. compare_and_bitpack_op.h
  101. compare_and_bitpack_op_gpu.cu.cc
  102. concat_lib.h
  103. concat_lib_cpu.cc
  104. concat_lib_cpu.h
  105. concat_lib_gpu.cc
  106. concat_lib_gpu.h
  107. concat_lib_gpu_impl.cu.cc
  108. concat_op.cc
  109. concat_op_test.cc
  110. conditional_accumulator.h
  111. conditional_accumulator_base.cc
  112. conditional_accumulator_base.h
  113. conditional_accumulator_base_op.cc
  114. conditional_accumulator_base_op.h
  115. conditional_accumulator_op.cc
  116. constant_op.cc
  117. constant_op.h
  118. constant_op_gpu.cu.cc
  119. constant_op_test.cc
  120. control_flow_ops.cc
  121. control_flow_ops.h
  122. control_flow_ops_test.cc
  123. conv_2d.h
  124. conv_2d_gpu.h
  125. conv_2d_gpu_double.cu.cc
  126. conv_2d_gpu_float.cu.cc
  127. conv_2d_gpu_half.cu.cc
  128. conv_2d_gpu_int.cu.cc
  129. conv_2d_gpu_int_spatial_convolution.cu.cc
  130. conv_2d_gpu_int_spatial_convolution_backward.cu.cc
  131. conv_2d_gpu_uint16.cu.cc
  132. conv_2d_gpu_uint32.cu.cc
  133. conv_2d_gpu_uint64.cu.cc
  134. conv_2d_gpu_uint8.cu.cc
  135. conv_3d.h
  136. conv_grad_filter_ops.cc
  137. conv_grad_filter_ops_benchmark_test.cc
  138. conv_grad_input_ops.cc
  139. conv_grad_input_ops_benchmark_test.cc
  140. conv_grad_ops.h
  141. conv_grad_ops_3d.cc
  142. conv_grad_shape_utils.cc
  143. conv_grad_shape_utils.h
  144. conv_ops.cc
  145. conv_ops.h
  146. conv_ops_3d.cc
  147. conv_ops_3d.h
  148. conv_ops_benchmark_test.cc
  149. conv_ops_fused_double.cc
  150. conv_ops_fused_float.cc
  151. conv_ops_fused_half.cc
  152. conv_ops_fused_image_transform.cc
  153. conv_ops_fused_impl.h
  154. conv_ops_gpu.h
  155. conv_ops_test.cc
  156. conv_ops_using_gemm.cc
  157. count_ops.cc
  158. count_ops_test.cc
  159. count_up_to_op.cc
  160. cross_op.cc
  161. cross_op.h
  162. cross_op_gpu.cu.cc
  163. cross_op_test.cc
  164. ctc_decoder_ops.cc
  165. ctc_loss_op.cc
  166. cudnn_pooling_gpu.cc
  167. cudnn_pooling_gpu.h
  168. cudnn_rnn_ops.cc
  169. cwise_op_abs.cc
  170. cwise_op_acos.cc
  171. cwise_op_acosh.cc
  172. cwise_op_add_1.cc
  173. cwise_op_add_2.cc
  174. cwise_op_arg.cc
  175. cwise_op_asin.cc
  176. cwise_op_asinh.cc
  177. cwise_op_atan.cc
  178. cwise_op_atan2.cc
  179. cwise_op_atanh.cc
  180. cwise_op_bitwise_and.cc
  181. cwise_op_bitwise_or.cc
  182. cwise_op_bitwise_xor.cc
  183. cwise_op_ceil.cc
  184. cwise_op_clip.cc
  185. cwise_op_clip.h
  186. cwise_op_clip_gpu.cu.cc
  187. cwise_op_complex.cc
  188. cwise_op_conj.cc
  189. cwise_op_cos.cc
  190. cwise_op_cosh.cc
  191. cwise_op_digamma.cc
  192. cwise_op_div.cc
  193. cwise_op_equal_to_1.cc
  194. cwise_op_equal_to_2.cc
  195. cwise_op_erf.cc
  196. cwise_op_erfc.cc
  197. cwise_op_exp.cc
  198. cwise_op_expm1.cc
  199. cwise_op_floor.cc
  200. cwise_op_floor_div.cc
  201. cwise_op_floor_mod.cc
  202. cwise_op_gpu_abs.cu.cc
  203. cwise_op_gpu_acos.cu.cc
  204. cwise_op_gpu_acosh.cu.cc
  205. cwise_op_gpu_add.cu.cc
  206. cwise_op_gpu_arg.cu.cc
  207. cwise_op_gpu_asin.cu.cc
  208. cwise_op_gpu_asinh.cu.cc
  209. cwise_op_gpu_atan.cu.cc
  210. cwise_op_gpu_atan2.cu.cc
  211. cwise_op_gpu_atanh.cu.cc
  212. cwise_op_gpu_bitwise_and.cu.cc
  213. cwise_op_gpu_bitwise_or.cu.cc
  214. cwise_op_gpu_bitwise_xor.cu.cc
  215. cwise_op_gpu_ceil.cu.cc
  216. cwise_op_gpu_complex.cu.cc
  217. cwise_op_gpu_conj.cu.cc
  218. cwise_op_gpu_cos.cu.cc
  219. cwise_op_gpu_cosh.cu.cc
  220. cwise_op_gpu_digamma.cu.cc
  221. cwise_op_gpu_div.cu.cc
  222. cwise_op_gpu_equal_to.cu.cc
  223. cwise_op_gpu_erf.cu.cc
  224. cwise_op_gpu_erfc.cu.cc
  225. cwise_op_gpu_exp.cu.cc
  226. cwise_op_gpu_expm1.cu.cc
  227. cwise_op_gpu_floor.cu.cc
  228. cwise_op_gpu_floor_div.cu.cc
  229. cwise_op_gpu_floor_mod.cu.cc
  230. cwise_op_gpu_greater.cu.cc
  231. cwise_op_gpu_greater_equal.cu.cc
  232. cwise_op_gpu_igammas_double.cu.cc
  233. cwise_op_gpu_igammas_float.cu.cc
  234. cwise_op_gpu_imag.cu.cc
  235. cwise_op_gpu_inverse.cu.cc
  236. cwise_op_gpu_invert.cu.cc
  237. cwise_op_gpu_isfinite.cu.cc
  238. cwise_op_gpu_isinf.cu.cc
  239. cwise_op_gpu_isnan.cu.cc
  240. cwise_op_gpu_left_shift.cu.cc
  241. cwise_op_gpu_less.cu.cc
  242. cwise_op_gpu_less_equal.cu.cc
  243. cwise_op_gpu_lgamma.cu.cc
  244. cwise_op_gpu_log.cu.cc
  245. cwise_op_gpu_log1p.cu.cc
  246. cwise_op_gpu_logical_and.cu.cc
  247. cwise_op_gpu_logical_not.cu.cc
  248. cwise_op_gpu_logical_or.cu.cc
  249. cwise_op_gpu_maximum.cu.cc
  250. cwise_op_gpu_minimum.cu.cc
  251. cwise_op_gpu_mod.cu.cc
  252. cwise_op_gpu_mul.cu.cc
  253. cwise_op_gpu_neg.cu.cc
  254. cwise_op_gpu_not_equal_to.cu.cc
  255. cwise_op_gpu_pow.cu.cc
  256. cwise_op_gpu_random_grad.cu.cc
  257. cwise_op_gpu_real.cu.cc
  258. cwise_op_gpu_right_shift.cu.cc
  259. cwise_op_gpu_rint.cu.cc
  260. cwise_op_gpu_round.cu.cc
  261. cwise_op_gpu_rsqrt.cu.cc
  262. cwise_op_gpu_select.cu.cc
  263. cwise_op_gpu_sigmoid.cu.cc
  264. cwise_op_gpu_sign.cu.cc
  265. cwise_op_gpu_sin.cu.cc
  266. cwise_op_gpu_sinh.cu.cc
  267. cwise_op_gpu_sqrt.cu.cc
  268. cwise_op_gpu_square.cu.cc
  269. cwise_op_gpu_squared_difference.cu.cc
  270. cwise_op_gpu_sub.cu.cc
  271. cwise_op_gpu_tan.cu.cc
  272. cwise_op_gpu_tanh.cu.cc
  273. cwise_op_gpu_xdivy.cu.cc
  274. cwise_op_gpu_xlog1py.cu.cc
  275. cwise_op_gpu_xlogy.cu.cc
  276. cwise_op_gpu_zeta.cu.cc
  277. cwise_op_greater.cc
  278. cwise_op_greater_equal.cc
  279. cwise_op_igammas.cc
  280. cwise_op_imag.cc
  281. cwise_op_invert.cc
  282. cwise_op_isfinite.cc
  283. cwise_op_isinf.cc
  284. cwise_op_isnan.cc
  285. cwise_op_left_shift.cc
  286. cwise_op_less.cc
  287. cwise_op_less_equal.cc
  288. cwise_op_lgamma.cc
  289. cwise_op_log.cc
  290. cwise_op_log1p.cc
  291. cwise_op_logical_and.cc
  292. cwise_op_logical_not.cc
  293. cwise_op_logical_or.cc
  294. cwise_op_maximum.cc
  295. cwise_op_minimum.cc
  296. cwise_op_mod.cc
  297. cwise_op_mul_1.cc
  298. cwise_op_mul_2.cc
  299. cwise_op_ndtri.cc
  300. cwise_op_ndtri_double.cu.cc
  301. cwise_op_ndtri_float.cu.cc
  302. cwise_op_neg_1.cc
  303. cwise_op_neg_2.cc
  304. cwise_op_not_equal_to_1.cc
  305. cwise_op_not_equal_to_2.cc
  306. cwise_op_pow.cc
  307. cwise_op_random_grad.cc
  308. cwise_op_real.cc
  309. cwise_op_reciprocal.cc
  310. cwise_op_right_shift.cc
  311. cwise_op_rint.cc
  312. cwise_op_round.cc
  313. cwise_op_rsqrt.cc
  314. cwise_op_select.cc
  315. cwise_op_sigmoid.cc
  316. cwise_op_sign.cc
  317. cwise_op_sin.cc
  318. cwise_op_sinh.cc
  319. cwise_op_sqrt.cc
  320. cwise_op_square.cc
  321. cwise_op_squared_difference.cc
  322. cwise_op_sub.cc
  323. cwise_op_tan.cc
  324. cwise_op_tanh.cc
  325. cwise_op_xdivy.cc
  326. cwise_op_xlog1py.cc
  327. cwise_op_xlogy.cc
  328. cwise_op_zeta.cc
  329. cwise_ops.h
  330. cwise_ops_common.cc
  331. cwise_ops_common.h
  332. cwise_ops_gpu_common.cu.h
  333. cwise_ops_gpu_gradients.cu.h
  334. cwise_ops_gradients.h
  335. cwise_ops_test.cc
  336. data_format_ops.cc
  337. data_format_ops.h
  338. data_format_ops_gpu.cu.cc
  339. debug_ops.cc
  340. debug_ops.h
  341. debug_ops_gpu.cu.cc
  342. debug_ops_test.cc
  343. decode_compressed_op.cc
  344. decode_csv_op.cc
  345. decode_padded_raw_op.cc
  346. decode_proto_op.cc
  347. decode_raw_op.cc
  348. decode_wav_op.cc
  349. decode_wav_op_test.cc
  350. deep_conv2d.cc
  351. deep_conv2d.h
  352. deep_conv2d_test.cc
  353. dense_update_functor.cc
  354. dense_update_functor.h
  355. dense_update_functor_gpu.cu.cc
  356. dense_update_ops.cc
  357. depthtospace_op.cc
  358. depthtospace_op.h
  359. depthtospace_op_gpu.cu.cc
  360. depthwise_conv_grad_op.cc
  361. depthwise_conv_op.cc
  362. depthwise_conv_op.h
  363. depthwise_conv_op_gpu.h
  364. depthwise_conv_op_gpu_double.cu.cc
  365. depthwise_conv_op_gpu_float.cu.cc
  366. depthwise_conv_op_gpu_half.cu.cc
  367. depthwise_conv_ops_test.cc
  368. dequantize_op.cc
  369. dequantize_op_test.cc
  370. deserialize_sparse_string_op.cc
  371. deserialize_sparse_variant_op.cc
  372. diag_op.cc
  373. diag_op.h
  374. diag_op_gpu.cu.cc
  375. diag_op_test.cc
  376. dilation_ops.cc
  377. dilation_ops.h
  378. dilation_ops_gpu.cu.cc
  379. dynamic_partition_op.cc
  380. dynamic_partition_op_gpu.cu.cc
  381. dynamic_partition_op_test.cc
  382. dynamic_stitch_op.cc
  383. dynamic_stitch_op_gpu.cu.cc
  384. dynamic_stitch_op_test.cc
  385. edit_distance_op.cc
  386. eigen_activations.h
  387. eigen_activations_test.cc
  388. eigen_attention.h
  389. eigen_attention_test.cc
  390. eigen_backward_cuboid_convolutions.h
  391. eigen_backward_cuboid_convolutions_test.cc
  392. eigen_backward_spatial_convolutions.h
  393. eigen_backward_spatial_convolutions_test.cc
  394. eigen_benchmark.h
  395. eigen_benchmark_cpu_test.cc
  396. eigen_contraction_kernel.cc
  397. eigen_contraction_kernel.h
  398. eigen_convolution_helpers.h
  399. eigen_cuboid_convolution.h
  400. eigen_mkldnn_contraction_kernel_test.cc
  401. eigen_pooling.h
  402. eigen_pooling_test.cc
  403. eigen_spatial_convolutions-inl.h
  404. eigen_spatial_convolutions.h
  405. eigen_spatial_convolutions_test.cc
  406. eigen_volume_patch.h
  407. encode_proto_op.cc
  408. encode_wav_op.cc
  409. encode_wav_op_test.cc
  410. example_parsing_ops.cc
  411. example_parsing_ops_test.cc
  412. fact_op.cc
  413. fake_quant_ops.cc
  414. fake_quant_ops_functor.h
  415. fake_quant_ops_gpu.cu.cc
  416. fake_quant_ops_test.cc
  417. fft_ops.cc
  418. fifo_queue.cc
  419. fifo_queue.h
  420. fifo_queue_op.cc
  421. fill_functor.cc
  422. fill_functor.cu.cc
  423. fill_functor.h
  424. fingerprint_op.cc
  425. fingerprint_op_test.cc
  426. fixed_length_record_reader_op.cc
  427. fractional_avg_pool_op.cc
  428. fractional_max_pool_op.cc
  429. fractional_pool_common.cc
  430. fractional_pool_common.h
  431. function_ops.cc
  432. function_ops.h
  433. functional_ops.cc
  434. fused_batch_norm_ex_op_test.cc
  435. fused_batch_norm_op.cc
  436. fused_batch_norm_op.cu.cc
  437. fused_batch_norm_op.h
  438. fused_batch_norm_op_test.cc
  439. fused_eigen_output_kernels.cc
  440. fused_eigen_output_kernels.h
  441. gather_functor.cc
  442. gather_functor.h
  443. gather_functor_batched.cc
  444. gather_functor_batched.h
  445. gather_functor_batched_gpu.cu.cc
  446. gather_functor_batched_gpu.cu.h
  447. gather_functor_gpu.cu.cc
  448. gather_functor_gpu.cu.h
  449. gather_nd_op.cc
  450. gather_nd_op.h
  451. gather_nd_op_cpu_impl.h
  452. gather_nd_op_cpu_impl_0.cc
  453. gather_nd_op_cpu_impl_1.cc
  454. gather_nd_op_cpu_impl_2.cc
  455. gather_nd_op_cpu_impl_3.cc
  456. gather_nd_op_cpu_impl_4.cc
  457. gather_nd_op_cpu_impl_5.cc
  458. gather_nd_op_cpu_impl_6.cc
  459. gather_nd_op_cpu_impl_7.cc
  460. gather_nd_op_gpu.cu.cc
  461. gather_nd_op_test.cc
  462. gather_op.cc
  463. gather_op_test.cc
  464. gemm_functors.h
  465. generate_vocab_remapping_op.cc
  466. gpu_device_array.h
  467. gpu_device_array_gpu.h
  468. gpu_prim.h
  469. gpu_utils.cc
  470. gpu_utils.h
  471. guarantee_const_op.cc
  472. guarantee_const_op_test.cc
  473. hinge-loss.h
  474. histogram_op.cc
  475. histogram_op.h
  476. histogram_op_gpu.cu.cc
  477. host_constant_op.cc
  478. host_constant_op.h
  479. i_remote_fused_graph_executor.h
  480. i_remote_fused_graph_ops_definitions.cc
  481. i_remote_fused_graph_ops_definitions.h
  482. identity_n_op.cc
  483. identity_n_op.h
  484. identity_n_op_test.cc
  485. identity_op.cc
  486. identity_op.h
  487. identity_op_test.cc
  488. identity_reader_op.cc
  489. immutable_constant_op.cc
  490. immutable_constant_op.h
  491. immutable_constant_op_test.cc
  492. in_topk_op.cc
  493. in_topk_op.h
  494. in_topk_op_gpu.cu.cc
  495. in_topk_op_test.cc
  496. initializable_lookup_table.cc
  497. initializable_lookup_table.h
  498. inplace_ops.cc
  499. inplace_ops_functor.h
  500. inplace_ops_functor_gpu.cu.cc
  501. isotonic_regression_op.cc
  502. isotonic_regression_op_test.cc
  503. kernel_platform_strings.h
  504. l2loss_op.cc
  505. l2loss_op.h
  506. l2loss_op_gpu.cu.cc
  507. linalg_ops_common.h
  508. list_kernels.cc
  509. list_kernels.cu.cc
  510. list_kernels.h
  511. listdiff_op.cc
  512. lmdb_reader_op.cc
  513. load_and_remap_matrix_op.cc
  514. logging_ops.cc
  515. logging_ops.h
  516. logging_ops_test.cc
  517. logistic-loss.h
  518. lookup_table_init_op.cc
  519. lookup_table_init_op.h
  520. lookup_table_op.cc
  521. lookup_table_op.h
  522. lookup_util.cc
  523. lookup_util.h
  524. loss.h
  525. loss_test.cc
  526. lrn_op.cc
  527. lrn_op_test.cc
  528. map_kernels.cc
  529. map_kernels.h
  530. map_stage_op.cc
  531. matching_files_op.cc
  532. matmul_op.cc
  533. matmul_op.h
  534. matmul_op_fused.cc
  535. matmul_op_test.cc
  536. maxpooling_op.cc
  537. maxpooling_op.h
  538. maxpooling_op_gpu.cu.cc
  539. maxpooling_op_gpu.h
  540. merge_v2_checkpoints_op_test.cc
  541. meta_support.cc
  542. meta_support.h
  543. mfcc.cc
  544. mfcc.h
  545. mfcc_dct.cc
  546. mfcc_dct.h
  547. mfcc_dct_test.cc
  548. mfcc_mel_filterbank.cc
  549. mfcc_mel_filterbank.h
  550. mfcc_mel_filterbank_test.cc
  551. mfcc_op.cc
  552. mfcc_op_test.cc
  553. mfcc_test.cc
  554. multinomial_op.cc
  555. multinomial_op.h
  556. multinomial_op_gpu.cu.cc
  557. multinomial_op_test.cc
  558. mutex_ops.cc
  559. nccl_ops.cc
  560. nextafter_op.cc
  561. nextafter_op.h
  562. nextafter_op_gpu.cu.cc
  563. nn_ops_test.cc
  564. no_op.cc
  565. no_op.h
  566. nth_element_op.cc
  567. nth_element_op.h
  568. one_hot_op.cc
  569. one_hot_op.h
  570. one_hot_op_gpu.cu.cc
  571. one_hot_op_test.cc
  572. ops_testutil.cc
  573. ops_testutil.h
  574. ops_testutil_test.cc
  575. ops_util.h
  576. ops_util_test.cc
  577. pack_op.cc
  578. pad_op.cc
  579. pad_op.h
  580. pad_op_gpu.cu.cc
  581. padding_fifo_queue.cc
  582. padding_fifo_queue.h
  583. padding_fifo_queue_op.cc
  584. parameterized_truncated_normal_op.cc
  585. parameterized_truncated_normal_op.h
  586. parameterized_truncated_normal_op_gpu.cu.cc
  587. parameterized_truncated_normal_op_test.cc
  588. parse_tensor_op.cc
  589. parse_tensor_test.cc
  590. partitioned_function_ops.cc
  591. partitioned_function_ops.h
  592. poisson-loss.h
  593. pooling_ops_3d.cc
  594. pooling_ops_3d.h
  595. pooling_ops_3d_gpu.cu.cc
  596. pooling_ops_3d_gpu.h
  597. pooling_ops_common.cc
  598. pooling_ops_common.h
  599. pooling_ops_common_gpu.h
  600. population_count_op.cc
  601. population_count_op.h
  602. population_count_op_gpu.cu.cc
  603. priority_queue.cc
  604. priority_queue.h
  605. priority_queue_op.cc
  606. quantization_utils.cc
  607. quantization_utils.h
  608. quantization_utils_test.cc
  609. quantize_and_dequantize_op.cc
  610. quantize_and_dequantize_op.h
  611. quantize_and_dequantize_op_gpu.cu.cc
  612. quantize_and_dequantize_op_test.cc
  613. quantize_down_and_shrink_range.cc
  614. quantize_down_and_shrink_range_op_test.cc
  615. quantize_op.cc
  616. quantize_op_test.cc
  617. quantized_activation_ops.cc
  618. quantized_activation_ops_test.cc
  619. quantized_add_op.cc
  620. quantized_add_op_test.cc
  621. quantized_batch_norm_op.cc
  622. quantized_batch_norm_op_test.cc
  623. quantized_bias_add_op.cc
  624. quantized_bias_add_op_test.cc
  625. quantized_concat_op.cc
  626. quantized_concat_op_test.cc
  627. quantized_conv_ops.cc
  628. quantized_conv_ops_test.cc
  629. quantized_instance_norm.cc
  630. quantized_instance_norm_test.cc
  631. quantized_matmul_op.cc
  632. quantized_matmul_op_test.cc
  633. quantized_mul_op.cc
  634. quantized_mul_op_test.cc
  635. quantized_pooling_ops.cc
  636. quantized_pooling_ops_test.cc
  637. quantized_reshape_op.cc
  638. quantized_reshape_op_test.cc
  639. quantized_resize_bilinear_op.cc
  640. quantized_resize_bilinear_op_test.cc
  641. queue_base.cc
  642. queue_base.h
  643. queue_op.cc
  644. queue_op.h
  645. queue_ops.cc
  646. ragged_cross_op.cc
  647. ragged_gather_op.cc
  648. ragged_gather_op_test.cc
  649. ragged_range_op.cc
  650. ragged_range_op_test.cc
  651. ragged_tensor_from_variant_op.cc
  652. ragged_tensor_from_variant_op_test.cc
  653. ragged_tensor_to_sparse_kernel.cc
  654. ragged_tensor_to_sparse_kernel_test.cc
  655. ragged_tensor_to_tensor_op.cc
  656. ragged_tensor_to_tensor_op_test.cc
  657. ragged_tensor_to_variant_op.cc
  658. ragged_tensor_to_variant_op_test.cc
  659. ragged_tensor_variant.cc
  660. ragged_tensor_variant.h
  661. random_binomial_op.cc
  662. random_binomial_op.h
  663. random_binomial_op_test.cc
  664. random_op.cc
  665. random_op.h
  666. random_op_cpu.h
  667. random_op_gpu.cu.cc
  668. random_op_gpu.h
  669. random_op_test.cc
  670. random_ops_util.h
  671. random_poisson_op.cc
  672. random_poisson_op.h
  673. random_poisson_op_test.cc
  674. random_shuffle_op.cc
  675. random_shuffle_queue_op.cc
  676. range_sampler.cc
  677. range_sampler.h
  678. range_sampler_test.cc
  679. reader_ops.cc
  680. record_input_op.cc
  681. record_yielder.cc
  682. record_yielder.h
  683. reduce_join_op.cc
  684. reduction_gpu_kernels.cu.h
  685. reduction_ops.h
  686. reduction_ops_all.cc
  687. reduction_ops_any.cc
  688. reduction_ops_common.cc
  689. reduction_ops_common.h
  690. reduction_ops_common_gpu.h
  691. reduction_ops_euclidean.cc
  692. reduction_ops_gpu_bool.cu.cc
  693. reduction_ops_gpu_complex128.cu.cc
  694. reduction_ops_gpu_complex64.cu.cc
  695. reduction_ops_gpu_double.cu.cc
  696. reduction_ops_gpu_float.cu.cc
  697. reduction_ops_gpu_int.cu.cc
  698. reduction_ops_half_mean_sum.cu.cc
  699. reduction_ops_half_prod_max_min.cu.cc
  700. reduction_ops_max.cc
  701. reduction_ops_mean.cc
  702. reduction_ops_min.cc
  703. reduction_ops_prod.cc
  704. reduction_ops_sum.cc
  705. reduction_ops_test.cc
  706. redux_functor.h
  707. reference_gemm.h
  708. regex_full_match_op.cc
  709. regex_replace_op.cc
  710. regex_replace_op_test.cc
  711. relu_op.cc
  712. relu_op.h
  713. relu_op_functor.h
  714. relu_op_gpu.cu.cc
  715. remote_fused_graph_execute_op.cc
  716. remote_fused_graph_execute_op_test.cc
  717. remote_fused_graph_execute_op_test_utils.cc
  718. remote_fused_graph_execute_op_test_utils.h
  719. remote_fused_graph_execute_utils.cc
  720. remote_fused_graph_execute_utils.h
  721. remote_fused_graph_execute_utils_test.cc
  722. remote_fused_graph_rewriter_transform.cc
  723. remote_fused_graph_rewriter_transform_test.cc
  724. requantization_range_op.cc
  725. requantization_range_op_test.cc
  726. requantize.cc
  727. requantize_op_test.cc
  728. reshape_op.cc
  729. reshape_op.h
  730. reshape_util.cc
  731. reshape_util.h
  732. resource_variable_ops.cc
  733. resource_variable_ops.h
  734. restore_op.cc
  735. restore_op_test.cc
  736. restore_v2_op_test.cc
  737. reverse_op.cc
  738. reverse_op.h
  739. reverse_op_gpu.cu.cc
  740. reverse_op_test.cc
  741. reverse_sequence_op.cc
  742. reverse_sequence_op.h
  743. reverse_sequence_op_gpu.cu.cc
  744. roll_op.cc
  745. roll_op.h
  746. roll_op_gpu.cu.cc
  747. roll_op_test.cc
  748. rpc_op.cc
  749. save_op.cc
  750. save_op_test.cc
  751. save_restore_tensor.cc
  752. save_restore_tensor.h
  753. save_restore_v2_ops.cc
  754. save_v2_op_test.cc
  755. scan_ops.cc
  756. scan_ops.h
  757. scan_ops_gpu.h
  758. scan_ops_gpu_double.cu.cc
  759. scan_ops_gpu_float.cu.cc
  760. scan_ops_gpu_half.cu.cc
  761. scan_ops_gpu_int.cu.cc
  762. scan_ops_test.cc
  763. scatter_functor.cc
  764. scatter_functor.h
  765. scatter_functor_gpu.cu.cc
  766. scatter_functor_gpu.cu.h
  767. scatter_nd_op.cc
  768. scatter_nd_op.h
  769. scatter_nd_op_cpu_impl.h
  770. scatter_nd_op_cpu_impl_0.cc
  771. scatter_nd_op_cpu_impl_1.cc
  772. scatter_nd_op_cpu_impl_2.cc
  773. scatter_nd_op_cpu_impl_3.cc
  774. scatter_nd_op_cpu_impl_4.cc
  775. scatter_nd_op_cpu_impl_5.cc
  776. scatter_nd_op_cpu_impl_6.cc
  777. scatter_nd_op_cpu_impl_7.cc
  778. scatter_nd_op_gpu.cu.cc
  779. scatter_nd_op_test.cc
  780. scatter_op.cc
  781. scatter_op_gpu.cu.cc
  782. scatter_op_test.cc
  783. scoped_allocator_ops.cc
  784. scoped_allocator_ops_test.cc
  785. sdca_internal.cc
  786. sdca_internal.h
  787. sdca_ops.cc
  788. sdca_ops_test.cc
  789. searchsorted_op.cc
  790. searchsorted_op.h
  791. searchsorted_op_gpu.cu.cc
  792. segment_reduction_ops.h
  793. segment_reduction_ops_gpu.cu.cc
  794. segment_reduction_ops_impl.h
  795. segment_reduction_ops_impl_1.cc
  796. segment_reduction_ops_impl_2.cc
  797. segment_reduction_ops_impl_3.cc
  798. segment_reduction_ops_impl_4.cc
  799. segment_reduction_ops_impl_5.cc
  800. segment_reduction_ops_test.cc
  801. sendrecv_ops.cc
  802. sendrecv_ops.h
  803. sendrecv_ops_test.cc
  804. sequence_ops.cc
  805. sequence_ops_test.cc
  806. serialize_sparse_op.cc
  807. session_ops.cc
  808. set_kernels.cc
  809. shape_ops.cc
  810. shape_ops.h
  811. shape_ops_test.cc
  812. slice_op.cc
  813. slice_op.h
  814. slice_op_cpu_impl.h
  815. slice_op_cpu_impl_1.cc
  816. slice_op_cpu_impl_2.cc
  817. slice_op_cpu_impl_3.cc
  818. slice_op_cpu_impl_4.cc
  819. slice_op_cpu_impl_5.cc
  820. slice_op_cpu_impl_6.cc
  821. slice_op_cpu_impl_7.cc
  822. slice_op_cpu_impl_8.cc
  823. slice_op_gpu.cu.cc
  824. slice_op_test.cc
  825. smooth-hinge-loss.h
  826. snapshot_op.cc
  827. snapshot_op.h
  828. snapshot_op_gpu.cu.cc
  829. sobol_op.cc
  830. softmax_op.cc
  831. softmax_op_functor.h
  832. softmax_op_gpu.cu.cc
  833. softplus_op.cc
  834. softplus_op.h
  835. softplus_op_gpu.cu.cc
  836. softsign_op.cc
  837. softsign_op.h
  838. softsign_op_gpu.cu.cc
  839. spacetobatch_benchmark_test.cc
  840. spacetobatch_functor.cc
  841. spacetobatch_functor.h
  842. spacetobatch_functor_gpu.cu.cc
  843. spacetobatch_op.cc
  844. spacetodepth_op.cc
  845. spacetodepth_op.h
  846. spacetodepth_op_gpu.cu.cc
  847. sparse_add_grad_op.cc
  848. sparse_add_op.cc
  849. sparse_add_op_test.cc
  850. sparse_concat_op.cc
  851. sparse_conditional_accumulator.h
  852. sparse_conditional_accumulator_op.cc
  853. sparse_cross_op.cc
  854. sparse_dense_binary_op_shared.cc
  855. sparse_dense_binary_op_shared_test.cc
  856. sparse_fill_empty_rows_op.cc
  857. sparse_matmul_op.cc
  858. sparse_matmul_op.h
  859. sparse_matmul_op_test.cc
  860. sparse_reduce_op.cc
  861. sparse_reduce_sum_op_test.cc
  862. sparse_reorder_op.cc
  863. sparse_reshape_op.cc
  864. sparse_slice_grad_op.cc
  865. sparse_slice_op.cc
  866. sparse_softmax_op.cc
  867. sparse_sparse_binary_op_shared.cc
  868. sparse_split_op.cc
  869. sparse_tensor_dense_add_op.cc
  870. sparse_tensor_dense_add_op.h
  871. sparse_tensor_dense_matmul_op.cc
  872. sparse_tensor_dense_matmul_op.h
  873. sparse_tensor_dense_matmul_op_gpu.cu.cc
  874. sparse_tensor_dense_matmul_op_test.cc
  875. sparse_tensors_map_ops.cc
  876. sparse_to_dense_op.cc
  877. sparse_to_dense_op_test.cc
  878. sparse_utils.cc
  879. sparse_utils.h
  880. sparse_utils_test.cc
  881. sparse_xent_op.cc
  882. sparse_xent_op.h
  883. sparse_xent_op_gpu.cu.cc
  884. sparse_xent_op_test.cc
  885. spectrogram.cc
  886. spectrogram.h
  887. spectrogram_convert_test_data.cc
  888. spectrogram_op.cc
  889. spectrogram_op_test.cc
  890. spectrogram_test.cc
  891. spectrogram_test_utils.cc
  892. spectrogram_test_utils.h
  893. split_lib.h
  894. split_lib_cpu.cc
  895. split_lib_gpu.cu.cc
  896. split_lib_gpu.h
  897. split_op.cc
  898. split_op_test.cc
  899. split_v_op.cc
  900. split_v_op_test.cc
  901. squared-loss.h
  902. stack.cc
  903. stack.h
  904. stack_ops.cc
  905. stage_op.cc
  906. stateful_random_ops.cc
  907. stateful_random_ops.h
  908. stateful_random_ops_cpu_gpu.h
  909. stateful_random_ops_gpu.cu.cc
  910. stateless_random_ops.cc
  911. stateless_random_ops.h
  912. stateless_random_ops_v2.cc
  913. stateless_random_ops_v2.h
  914. strided_slice_op.cc
  915. strided_slice_op.h
  916. strided_slice_op_define_grad.cc
  917. strided_slice_op_gpu_bool.cu.cc
  918. strided_slice_op_gpu_complex.cu.cc
  919. strided_slice_op_gpu_impl.h
  920. strided_slice_op_gpu_int.cu.cc
  921. strided_slice_op_gpu_number_types.cu.cc
  922. strided_slice_op_impl.h
  923. strided_slice_op_inst_0.cc
  924. strided_slice_op_inst_1.cc
  925. strided_slice_op_inst_2.cc
  926. strided_slice_op_inst_3.cc
  927. strided_slice_op_inst_4.cc
  928. strided_slice_op_inst_5.cc
  929. strided_slice_op_inst_6.cc
  930. strided_slice_op_inst_7.cc
  931. strided_slice_op_inst_8.cc
  932. strided_slice_op_test.cc
  933. string_format_op.cc
  934. string_format_op_test.cc
  935. string_join_op.cc
  936. string_length_op.cc
  937. string_lower_op.cc
  938. string_ngrams_op.cc
  939. string_ngrams_op_test.cc
  940. string_split_op.cc
  941. string_split_op_test.cc
  942. string_strip_op.cc
  943. string_to_hash_bucket_op.cc
  944. string_to_hash_bucket_op.h
  945. string_to_number_op.cc
  946. string_upper_op.cc
  947. string_util.cc
  948. string_util.h
  949. substr_op.cc
  950. substr_op_test.cc
  951. summary_audio_op.cc
  952. summary_audio_op_test.cc
  953. summary_image_op.cc
  954. summary_image_op_test.cc
  955. summary_interface.h
  956. summary_kernels.cc
  957. summary_op_test.cc
  958. summary_tensor_op.cc
  959. summary_tensor_op_test.cc
  960. tensor_array.cc
  961. tensor_array.h
  962. tensor_array_ops.cc
  963. tensor_cord.cc
  964. tensor_cord.h
  965. tensor_cord_test.cc
  966. tensor_flag_utils.cc
  967. tensor_flag_utils.h
  968. tensor_flag_utils_test.cc
  969. tensor_list.cc
  970. tensor_list.h
  971. tensor_map.cc
  972. tensor_map.h
  973. tensor_map_test.cc
  974. text_line_reader_op.cc
  975. tf_record_reader_op.cc
  976. tile_functor.h
  977. tile_functor_cpu.h
  978. tile_functor_cpu_bfloat16.cc
  979. tile_functor_cpu_bool.cc
  980. tile_functor_cpu_complex128.cc
  981. tile_functor_cpu_complex64.cc
  982. tile_functor_cpu_double.cc
  983. tile_functor_cpu_float.cc
  984. tile_functor_cpu_half.cc
  985. tile_functor_cpu_int16.cc
  986. tile_functor_cpu_int32.cc
  987. tile_functor_cpu_int64.cc
  988. tile_functor_cpu_int8.cc
  989. tile_functor_cpu_tstring.cc
  990. tile_functor_cpu_uint32.cc
  991. tile_functor_cpu_uint64.cc
  992. tile_functor_cpu_uint8.cc
  993. tile_functor_cpu_variant.cc
  994. tile_functor_gpu.h
  995. tile_functor_gpu_bool.cu.cc
  996. tile_functor_gpu_complex128.cu.cc
  997. tile_functor_gpu_complex64.cu.cc
  998. tile_functor_gpu_double.cu.cc
  999. tile_functor_gpu_float.cu.cc
  1000. tile_functor_gpu_half.cu.cc
  1001. tile_functor_gpu_int16.cu.cc
  1002. tile_functor_gpu_int32.cu.cc
  1003. tile_functor_gpu_int64.cu.cc
  1004. tile_functor_sycl.cc
  1005. tile_ops.cc
  1006. tile_ops_cpu_impl.h
  1007. tile_ops_cpu_impl_1.cc
  1008. tile_ops_cpu_impl_2.cc
  1009. tile_ops_cpu_impl_3.cc
  1010. tile_ops_cpu_impl_4.cc
  1011. tile_ops_cpu_impl_5.cc
  1012. tile_ops_cpu_impl_6.cc
  1013. tile_ops_cpu_impl_7.cc
  1014. tile_ops_gpu_impl.h
  1015. tile_ops_gpu_impl_1.cu.cc
  1016. tile_ops_gpu_impl_2.cu.cc
  1017. tile_ops_gpu_impl_3.cu.cc
  1018. tile_ops_gpu_impl_4.cu.cc
  1019. tile_ops_gpu_impl_5.cu.cc
  1020. tile_ops_gpu_impl_6.cu.cc
  1021. tile_ops_gpu_impl_7.cu.cc
  1022. tile_ops_gpu_impl_8.cu.cc
  1023. tile_ops_impl.h
  1024. topk_op.cc
  1025. topk_op.h
  1026. topk_op_gpu.h
  1027. topk_op_gpu_double.cu.cc
  1028. topk_op_gpu_float.cu.cc
  1029. topk_op_gpu_half.cu.cc
  1030. topk_op_gpu_int16.cu.cc
  1031. topk_op_gpu_int32.cu.cc
  1032. topk_op_gpu_int64.cu.cc
  1033. topk_op_gpu_int8.cu.cc
  1034. topk_op_gpu_uint16.cu.cc
  1035. topk_op_gpu_uint32.cu.cc
  1036. topk_op_gpu_uint64.cu.cc
  1037. topk_op_gpu_uint8.cu.cc
  1038. training_op_helpers.cc
  1039. training_op_helpers.h
  1040. training_ops.cc
  1041. training_ops.h
  1042. training_ops_gpu.cu.cc
  1043. training_ops_test.cc
  1044. transpose_functor.h
  1045. transpose_functor_cpu.cc
  1046. transpose_functor_gpu.cu.cc
  1047. transpose_op.cc
  1048. transpose_op.h
  1049. transpose_util_test.cc
  1050. typed_conditional_accumulator_base.h
  1051. typed_queue.h
  1052. unary_ops_composition.cc
  1053. unary_ops_composition_test.cc
  1054. unicode_ops.cc
  1055. unicode_script_op.cc
  1056. unique_op.cc
  1057. unique_op_test.cc
  1058. unpack_op.cc
  1059. unravel_index_op.cc
  1060. unsorted_segment_join_op.cc
  1061. variable_ops.cc
  1062. variable_ops.h
  1063. variable_ops_test.cc
  1064. where_op.cc
  1065. where_op.h
  1066. where_op_gpu.cu.h
  1067. where_op_gpu_impl_1.cu.cc
  1068. where_op_gpu_impl_2.cu.cc
  1069. where_op_gpu_impl_3.cu.cc
  1070. where_op_gpu_impl_4.cu.cc
  1071. where_op_gpu_impl_5.cu.cc
  1072. where_op_gpu_impl_6.cu.cc
  1073. where_op_gpu_impl_7.cu.cc
  1074. where_op_gpu_impl_8.cu.cc
  1075. whole_file_read_ops.cc
  1076. winograd_transform.h
  1077. word2vec_kernels.cc
  1078. xent_op.cc
  1079. xent_op.h
  1080. xent_op_gpu.cu.cc
  1081. xent_op_test.cc
  1082. xsmm_conv2d.cc
  1083. xsmm_conv2d.h
  1084. xsmm_conv2d_test.cc