Home
last modified time | relevance | path

Searched refs:DECLARE_GPU_SPECS (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dgather_functor.cc36 #define DECLARE_GPU_SPECS(T) \ macro
40 TF_CALL_int64(DECLARE_GPU_SPECS);
41 TF_CALL_GPU_NUMBER_TYPES(DECLARE_GPU_SPECS);
42 TF_CALL_COMPLEX_TYPES(DECLARE_GPU_SPECS);
44 #undef DECLARE_GPU_SPECS
Dgather_functor_batched.cc36 #define DECLARE_GPU_SPECS(T) \ macro
40 TF_CALL_int64(DECLARE_GPU_SPECS);
41 TF_CALL_GPU_NUMBER_TYPES(DECLARE_GPU_SPECS);
42 TF_CALL_COMPLEX_TYPES(DECLARE_GPU_SPECS);
44 #undef DECLARE_GPU_SPECS
Dgather_nd_op.cc101 #define DECLARE_GPU_SPECS(T) \ macro
105 TF_CALL_int32(DECLARE_GPU_SPECS);
106 TF_CALL_int64(DECLARE_GPU_SPECS);
107 TF_CALL_GPU_NUMBER_TYPES(DECLARE_GPU_SPECS);
108 TF_CALL_COMPLEX_TYPES(DECLARE_GPU_SPECS);
110 #undef DECLARE_GPU_SPECS
Dscatter_nd_op_gpu.cu.cc189 #define DECLARE_GPU_SPECS(T) \ macro
197 TF_CALL_int32(DECLARE_GPU_SPECS);
199 TF_CALL_int64(DECLARE_GPU_SPECS);
201 TF_CALL_GPU_NUMBER_TYPES(DECLARE_GPU_SPECS);
203 TF_CALL_COMPLEX_TYPES(DECLARE_GPU_SPECS);
205 #undef DECLARE_GPU_SPECS
Dscatter_functor.cc54 #define DECLARE_GPU_SPECS(T) \ macro
58 TF_CALL_GPU_NUMBER_TYPES(DECLARE_GPU_SPECS);
60 #undef DECLARE_GPU_SPECS
Dbatch_norm_op.cc188 #define DECLARE_GPU_SPECS(T) DECLARE_GPU_SPEC(T); macro
190 TF_CALL_half(DECLARE_GPU_SPECS);
191 TF_CALL_float(DECLARE_GPU_SPECS);
236 #define DECLARE_GPU_SPECS(T) DECLARE_GPU_SPEC(T); macro
238 TF_CALL_half(DECLARE_GPU_SPECS);
239 TF_CALL_float(DECLARE_GPU_SPECS);
Ddata_format_ops.cc264 #define DECLARE_GPU_SPECS(T) DECLARE_GPU_SPEC(T); macro
265 TF_CALL_int32(DECLARE_GPU_SPECS);
266 TF_CALL_int64(DECLARE_GPU_SPECS);
276 #define DECLARE_GPU_SPECS(T) DECLARE_GPU_SPEC(T); macro
277 TF_CALL_int32(DECLARE_GPU_SPECS);
278 TF_CALL_int64(DECLARE_GPU_SPECS);
Dargmax_op.cc175 #define DECLARE_GPU_SPECS(T) \ macro
197 TF_CALL_GPU_NUMBER_TYPES(DECLARE_GPU_SPECS);
198 TF_CALL_bool(DECLARE_GPU_SPECS);
202 #undef DECLARE_GPU_SPECS
Dreverse_sequence_op.cc199 #define DECLARE_GPU_SPECS(T) \ macro
205 TF_CALL_GPU_NUMBER_TYPES(DECLARE_GPU_SPECS);
206 TF_CALL_bool(DECLARE_GPU_SPECS);
Dpad_op.cc306 #define DECLARE_GPU_SPECS(T) \ macro
315 TF_CALL_GPU_ALL_TYPES(DECLARE_GPU_SPECS);
316 TF_CALL_int8(DECLARE_GPU_SPECS);
317 TF_CALL_uint8(DECLARE_GPU_SPECS);
Dscatter_nd_op.cc1015 #define DECLARE_GPU_SPECS(T) \ macro
1023 TF_CALL_int32(DECLARE_GPU_SPECS);
1025 TF_CALL_GPU_NUMBER_TYPES(DECLARE_GPU_SPECS);
1027 TF_CALL_COMPLEX_TYPES(DECLARE_GPU_SPECS);
1030 #undef DECLARE_GPU_SPECS
Dcompare_and_bitpack_op_gpu.cu.cc134 #undef DECLARE_GPU_SPECS
/external/tensorflow/tensorflow/core/kernels/linalg/
Deinsum_op_gpu.cu.cc28 #define DECLARE_GPU_SPECS(T) \ macro
36 TF_CALL_GPU_NUMBER_TYPES(DECLARE_GPU_SPECS);
37 TF_CALL_COMPLEX_TYPES(DECLARE_GPU_SPECS);
40 #undef DECLARE_GPU_SPECS
Deinsum_op_impl.h758 #define DECLARE_GPU_SPECS(T) \ macro
766 DECLARE_GPU_SPECS(Eigen::half);
767 DECLARE_GPU_SPECS(double);
768 DECLARE_GPU_SPECS(float);
771 DECLARE_GPU_SPECS(complex64);
772 DECLARE_GPU_SPECS(complex128);
775 #undef DECLARE_GPU_SPECS
/external/tensorflow/tensorflow/core/kernels/image/
Dmirror_pad_op.cc212 #define DECLARE_GPU_SPECS(T) \ macro
224 TF_CALL_GPU_NUMBER_TYPES(DECLARE_GPU_SPECS);
225 #undef DECLARE_GPU_SPECS
419 #define DECLARE_GPU_SPECS(T) \ macro
431 TF_CALL_GPU_NUMBER_TYPES(DECLARE_GPU_SPECS);
432 #undef DECLARE_GPU_SPECS
/external/tensorflow/tensorflow/core/kernels/rnn/
Dlstm_ops.cc288 #define DECLARE_GPU_SPECS(T) DECLARE_GPU_FBPROP(T, ICFO); macro
290 DECLARE_GPU_SPECS(float);
291 DECLARE_GPU_SPECS(Eigen::half);
292 #undef DECLARE_GPU_SPECS
1034 #define DECLARE_GPU_SPECS(T) \ macro
1047 DECLARE_GPU_SPECS(Eigen::half);
1048 DECLARE_GPU_SPECS(float);
1049 #undef DECLARE_GPU_SPECS
1342 #define DECLARE_GPU_SPECS(T) \ macro
1369 DECLARE_GPU_SPECS(Eigen::half);
[all …]
Dlstm_ops_gpu.cu.cc463 #define DECLARE_GPU_SPECS(T) \ macro
474 DECLARE_GPU_SPECS(Eigen::half);
475 DECLARE_GPU_SPECS(float);
476 #undef DECLARE_GPU_SPECS