Home
last modified time | relevance | path

Searched defs:DECLARE_GPU_SPEC (Results 1 – 25 of 38) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/sparse/
Dsoftmax_op.cc100 #define DECLARE_GPU_SPEC(T) \ macro
208 #define DECLARE_GPU_SPEC(T) \ macro
Dsparse_matrix_components_op.cc131 #define DECLARE_GPU_SPEC(T) \ macro
Dmul_op.cc153 #define DECLARE_GPU_SPEC(T) \ macro
/external/tensorflow/tensorflow/core/kernels/
Dbatch_norm_op.cc178 #define DECLARE_GPU_SPEC(T) \ macro
223 #define DECLARE_GPU_SPEC(T) \ macro
Dtraining_ops.cc1018 #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 …]
Ddata_format_ops.cc279 #define DECLARE_GPU_SPEC(T) \ macro
290 #define DECLARE_GPU_SPEC(T) \ macro
Din_topk_op.cc123 #define DECLARE_GPU_SPEC(T, TARGET_T) \ macro
Drelu_op.cc129 #define DECLARE_GPU_SPEC(T) \ macro
Dbetainc_op.cc147 #define DECLARE_GPU_SPEC(T) \ macro
Dpopulation_count_op.cc147 #define DECLARE_GPU_SPEC(T) \ macro
Dargmax_op.cc177 #define DECLARE_GPU_SPEC(T, Tout, Dims) \ macro
Done_hot_op.cc158 #define DECLARE_GPU_SPEC(T) \ macro
Dsparse_fill_empty_rows_op.cc310 #define DECLARE_GPU_SPEC(T, Tindex) \ macro
437 #define DECLARE_GPU_SPEC(T, Tindex) \ macro
Dreverse_sequence_op.cc189 #define DECLARE_GPU_SPEC(T, Tlen, Dims) \ macro
Dpad_op.cc294 #define DECLARE_GPU_SPEC(T, Dims) \ macro
Dtopk_op.cc257 #define DECLARE_GPU_SPEC(T) \ macro
Dreverse_op.cc335 #define DECLARE_GPU_SPEC(T) \ macro
/external/tensorflow/tensorflow/core/kernels/rnn/
Dgru_ops.cc388 #define DECLARE_GPU_SPEC(T) \ macro
418 #define DECLARE_GPU_SPEC(T) \ macro
/external/tensorflow/tensorflow/core/kernels/image/
Dmirror_pad_op.cc206 #define DECLARE_GPU_SPEC(T, Tpaddings, i) \ macro
412 #define DECLARE_GPU_SPEC(T, Tpaddings, k) \ macro
Dresize_nearest_neighbor_op_gpu.cu.cc192 #define DECLARE_GPU_SPEC(T) \ macro
246 #define DECLARE_GPU_SPEC(T) \ macro
Dextract_volume_patches_op.cc170 #define DECLARE_GPU_SPEC(T) \ macro
Dextract_image_patches_op.cc142 #define DECLARE_GPU_SPEC(T) \ macro
Dadjust_contrast_op.cc115 #define DECLARE_GPU_SPEC(T) \ macro
400 #define DECLARE_GPU_SPEC(T) \ macro
/external/tensorflow/tensorflow/core/kernels/linalg/
Dmatrix_band_part_op.cc204 #define DECLARE_GPU_SPEC(T) \ macro
Dmatrix_set_diag_op.cc266 #define DECLARE_GPU_SPEC(T) \ macro

12