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