Home
last modified time | relevance | path

Searched refs:DECLARE_GPU_SPECS (Results 1 – 12 of 12) 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_complex64(DECLARE_GPU_SPECS);
43 TF_CALL_complex128(DECLARE_GPU_SPECS);
45 #undef DECLARE_GPU_SPECS
Dgather_nd_op.cc228 #define DECLARE_GPU_SPECS(T) \ macro
232 TF_CALL_int32(DECLARE_GPU_SPECS);
233 TF_CALL_int64(DECLARE_GPU_SPECS);
234 TF_CALL_GPU_NUMBER_TYPES(DECLARE_GPU_SPECS);
235 TF_CALL_complex64(DECLARE_GPU_SPECS);
236 TF_CALL_complex128(DECLARE_GPU_SPECS);
238 #undef DECLARE_GPU_SPECS
Dscatter_nd_op_gpu.cu.cc169 #define DECLARE_GPU_SPECS(T) \ macro
173 TF_CALL_int32(DECLARE_GPU_SPECS);
174 TF_CALL_GPU_NUMBER_TYPES(DECLARE_GPU_SPECS);
175 TF_CALL_complex64(DECLARE_GPU_SPECS);
176 TF_CALL_complex128(DECLARE_GPU_SPECS);
178 #undef DECLARE_GPU_SPECS
Ddata_format_ops.cc178 #define DECLARE_GPU_SPECS(T) DECLARE_GPU_SPEC(T); macro
179 TF_CALL_int32(DECLARE_GPU_SPECS);
180 TF_CALL_int64(DECLARE_GPU_SPECS);
190 #define DECLARE_GPU_SPECS(T) DECLARE_GPU_SPEC(T); macro
191 TF_CALL_int32(DECLARE_GPU_SPECS);
192 TF_CALL_int64(DECLARE_GPU_SPECS);
Dscatter_functor.cc54 #define DECLARE_GPU_SPECS(T) \ macro
58 TF_CALL_GPU_NUMBER_TYPES_NO_HALF(DECLARE_GPU_SPECS);
60 #undef DECLARE_GPU_SPECS
Dbatch_norm_op.cc191 #define DECLARE_GPU_SPECS(T) DECLARE_GPU_SPEC(T); macro
193 TF_CALL_half(DECLARE_GPU_SPECS);
194 TF_CALL_float(DECLARE_GPU_SPECS);
250 #define DECLARE_GPU_SPECS(T) DECLARE_GPU_SPEC(T); macro
252 TF_CALL_half(DECLARE_GPU_SPECS);
253 TF_CALL_float(DECLARE_GPU_SPECS);
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
Dreverse_sequence_op.cc197 #define DECLARE_GPU_SPECS(T) \ macro
203 TF_CALL_GPU_NUMBER_TYPES(DECLARE_GPU_SPECS);
204 TF_CALL_bool(DECLARE_GPU_SPECS);
Dargmax_op.cc168 #define DECLARE_GPU_SPECS(T) \ macro
186 TF_CALL_GPU_NUMBER_TYPES(DECLARE_GPU_SPECS);
189 #undef DECLARE_GPU_SPECS
Dpad_op.cc314 #define DECLARE_GPU_SPECS(T) \ macro
323 TF_CALL_GPU_ALL_TYPES(DECLARE_GPU_SPECS);
324 TF_CALL_int8(DECLARE_GPU_SPECS);
325 TF_CALL_uint8(DECLARE_GPU_SPECS);
Dscatter_nd_op.cc766 #define DECLARE_GPU_SPECS(T) \ macro
770 TF_CALL_int32(DECLARE_GPU_SPECS);
771 TF_CALL_GPU_NUMBER_TYPES(DECLARE_GPU_SPECS);
772 TF_CALL_complex64(DECLARE_GPU_SPECS);
773 TF_CALL_complex128(DECLARE_GPU_SPECS);
775 #undef DECLARE_GPU_SPECS
Dcompare_and_bitpack_op_gpu.cu.cc136 #undef DECLARE_GPU_SPECS