| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | training_ops.cc | 1018 #define DECLARE_GPU_SPEC(T) \ macro
1162 #define DECLARE_GPU_SPEC(T) \ macro
1318 #define DECLARE_GPU_SPEC(T, Tindex) \ macro
1645 #define DECLARE_GPU_SPEC(T) \ macro
1746 #define DECLARE_GPU_SPEC(T) \ macro
1860 #define DECLARE_GPU_SPEC(T) \ macro
1981 #define DECLARE_GPU_SPEC(T, Tindex) \ macro
2114 #define DECLARE_GPU_SPEC(T, Tindex) \ macro
2260 #define DECLARE_GPU_SPEC(T, Tindex) \ macro
2722 #define DECLARE_GPU_SPEC(T) \ macro
[all …]
|
| H A D | batch_norm_op.cc | 178 #define DECLARE_GPU_SPEC(T) \ macro
223 #define DECLARE_GPU_SPEC(T) \ macro
|
| H A D | data_format_ops.cc | 279 #define DECLARE_GPU_SPEC(T) \ macro
290 #define DECLARE_GPU_SPEC(T) \ macro
|
| H A D | sparse_fill_empty_rows_op.cc | 310 #define DECLARE_GPU_SPEC(T, Tindex) \ macro
437 #define DECLARE_GPU_SPEC(T, Tindex) \ macro
|
| H A D | in_topk_op.cc | 123 #define DECLARE_GPU_SPEC(T, TARGET_T) \ macro
|
| H A D | relu_op.cc | 129 #define DECLARE_GPU_SPEC(T) \ macro
|
| H A D | betainc_op.cc | 147 #define DECLARE_GPU_SPEC(T) \ macro
|
| H A D | argmax_op.cc | 177 #define DECLARE_GPU_SPEC(T, Tout, Dims) \ macro
|
| H A D | population_count_op.cc | 147 #define DECLARE_GPU_SPEC(T) \ macro
|
| H A D | one_hot_op.cc | 158 #define DECLARE_GPU_SPEC(T) \ macro
|
| H A D | reverse_sequence_op.cc | 189 #define DECLARE_GPU_SPEC(T, Tlen, Dims) \ macro
|
| H A D | pad_op.cc | 294 #define DECLARE_GPU_SPEC(T, Dims) \ macro
|
| H A D | topk_op.cc | 257 #define DECLARE_GPU_SPEC(T) \ macro
|
| H A D | reverse_op.cc | 335 #define DECLARE_GPU_SPEC(T) \ macro
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/sparse/ |
| H A D | softmax_op.cc | 100 #define DECLARE_GPU_SPEC(T) \ macro
208 #define DECLARE_GPU_SPEC(T) \ macro
|
| H A D | sparse_matrix_components_op.cc | 131 #define DECLARE_GPU_SPEC(T) \ macro
|
| H A D | mul_op.cc | 153 #define DECLARE_GPU_SPEC(T) \ macro
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/rnn/ |
| H A D | gru_ops.cc | 388 #define DECLARE_GPU_SPEC(T) \ macro
418 #define DECLARE_GPU_SPEC(T) \ macro
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/image/ |
| H A D | mirror_pad_op.cc | 206 #define DECLARE_GPU_SPEC(T, Tpaddings, i) \ macro
412 #define DECLARE_GPU_SPEC(T, Tpaddings, k) \ macro
|
| H A D | resize_nearest_neighbor_op_gpu.cu.cc | 192 #define DECLARE_GPU_SPEC(T) \ macro
246 #define DECLARE_GPU_SPEC(T) \ macro
|
| H A D | adjust_contrast_op.cc | 115 #define DECLARE_GPU_SPEC(T) \ macro
400 #define DECLARE_GPU_SPEC(T) \ macro
|
| H A D | extract_image_patches_op.cc | 142 #define DECLARE_GPU_SPEC(T) \ macro
|
| H A D | extract_volume_patches_op.cc | 170 #define DECLARE_GPU_SPEC(T) \ macro
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/ |
| H A D | matrix_set_diag_op.cc | 266 #define DECLARE_GPU_SPEC(T) \ macro
|
| H A D | matrix_band_part_op.cc | 204 #define DECLARE_GPU_SPEC(T) \ macro
|