Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/kernels/sparse/
Dsoftmax_op.cc100 #define DECLARE_GPU_SPEC(T) \ macro
209 #define DECLARE_GPU_SPEC(T) \ macro
Dsparse_matrix_components_op.cc132 #define DECLARE_GPU_SPEC(T) \ macro
Dmul_op.cc155 #define DECLARE_GPU_SPEC(T) \ macro
/external/tensorflow/tensorflow/core/kernels/
Ddata_format_ops.cc182 #define DECLARE_GPU_SPEC(T) \ macro
193 #define DECLARE_GPU_SPEC(T) \ macro
Dbatch_norm_op.cc181 #define DECLARE_GPU_SPEC(T) \ macro
237 #define DECLARE_GPU_SPEC(T) \ macro
Drelu_op.cc80 #define DECLARE_GPU_SPEC(T) \ macro
Din_topk_op.cc123 #define DECLARE_GPU_SPEC(T, TARGET_T) \ macro
Dmirror_pad_op.cc204 #define DECLARE_GPU_SPEC(T, Tpaddings, i) \ macro
410 #define DECLARE_GPU_SPEC(T, Tpaddings, k) \ macro
Dsoftsign_op.cc94 #define DECLARE_GPU_SPEC(T) \ macro
Dsoftplus_op.cc94 #define DECLARE_GPU_SPEC(T) \ macro
Dbetainc_op.cc147 #define DECLARE_GPU_SPEC(T) \ macro
Dresize_nearest_neighbor_op_gpu.cu.cc194 #define DECLARE_GPU_SPEC(T) \ macro
251 #define DECLARE_GPU_SPEC(T) \ macro
Dpopulation_count_op.cc143 #define DECLARE_GPU_SPEC(T) \ macro
Dargmax_op.cc164 #define DECLARE_GPU_SPEC(T, Tout, Dims) \ macro
Done_hot_op.cc158 #define DECLARE_GPU_SPEC(T) \ macro
Dcholesky_op.cc79 #define DECLARE_GPU_SPEC(T) \ macro
Dextract_volume_patches_op.cc167 #define DECLARE_GPU_SPEC(T) \ macro
Dextract_image_patches_op.cc139 #define DECLARE_GPU_SPEC(T) \ macro
Dtraining_ops.cc661 #define DECLARE_GPU_SPEC(T) \ macro
836 #define DECLARE_GPU_SPEC(T) \ macro
1325 #define DECLARE_GPU_SPEC(T) \ macro
1438 #define DECLARE_GPU_SPEC(T) \ macro
2509 #define DECLARE_GPU_SPEC(T) \ macro
2553 #define DECLARE_GPU_SPEC(T) \ macro
2933 #define DECLARE_GPU_SPEC(T) \ macro
3167 #define DECLARE_GPU_SPEC(T) \ macro
3303 #define DECLARE_GPU_SPEC(T, Tindex) \ macro
3585 #define DECLARE_GPU_SPEC(T) \ macro
[all …]
Dcompare_and_bitpack_op.cc181 #define DECLARE_GPU_SPEC(T) \ macro
Dadjust_contrast_op.cc107 #define DECLARE_GPU_SPEC(T) \ macro
392 #define DECLARE_GPU_SPEC(T) \ macro
Dreverse_sequence_op.cc184 #define DECLARE_GPU_SPEC(T, Tlen, Dims) \ macro
Dmatrix_band_part_op.cc203 #define DECLARE_GPU_SPEC(T) \ macro
Dmatrix_set_diag_op.cc263 #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

12