| /external/tensorflow/tensorflow/core/kernels/ |
| D | gather_functor_gpu.cu.cc | 30 #define DEFINE_GPU_SPECS(T) \ macro
34 TF_CALL_bool(DEFINE_GPU_SPECS);
35 TF_CALL_int32(DEFINE_GPU_SPECS);
36 TF_CALL_int64(DEFINE_GPU_SPECS);
37 TF_CALL_GPU_NUMBER_TYPES(DEFINE_GPU_SPECS);
38 TF_CALL_complex64(DEFINE_GPU_SPECS);
39 TF_CALL_complex128(DEFINE_GPU_SPECS);
41 #undef DEFINE_GPU_SPECS
|
| D | population_count_op_gpu.cu.cc | 66 #define DEFINE_GPU_SPECS(T) \ macro
79 TF_CALL_uint8(DEFINE_GPU_SPECS);
80 TF_CALL_int8(DEFINE_GPU_SPECS);
81 TF_CALL_uint16(DEFINE_GPU_SPECS);
82 TF_CALL_int16(DEFINE_GPU_SPECS);
83 TF_CALL_int32(DEFINE_GPU_SPECS);
84 TF_CALL_int64(DEFINE_GPU_SPECS);
86 #undef DEFINE_GPU_SPECS
|
| D | scatter_op_gpu.cu.cc | 40 #define DEFINE_GPU_SPECS(T) \ macro
44 DEFINE_GPU_SPECS(Eigen::half);
45 DEFINE_GPU_SPECS(float);
46 DEFINE_GPU_SPECS(double);
50 #undef DEFINE_GPU_SPECS
|
| D | gather_nd_op_gpu.cu.cc | 117 #define DEFINE_GPU_SPECS(T) \ macro
121 TF_CALL_int32(DEFINE_GPU_SPECS);
122 TF_CALL_int64(DEFINE_GPU_SPECS);
123 TF_CALL_GPU_NUMBER_TYPES(DEFINE_GPU_SPECS);
124 TF_CALL_complex64(DEFINE_GPU_SPECS);
125 TF_CALL_complex128(DEFINE_GPU_SPECS);
127 #undef DEFINE_GPU_SPECS
|
| D | pad_op_gpu.cu.cc | 38 #define DEFINE_GPU_SPECS(T) \ macro
42 TF_CALL_GPU_ALL_TYPES(DEFINE_GPU_SPECS);
43 TF_CALL_int8(DEFINE_GPU_SPECS);
44 TF_CALL_uint8(DEFINE_GPU_SPECS);
|
| D | scatter_functor_gpu.cu.cc | 39 #define DEFINE_GPU_SPECS(T) \ macro
43 DEFINE_GPU_SPECS(float);
44 DEFINE_GPU_SPECS(double);
51 #undef DEFINE_GPU_SPECS
|
| D | reverse_sequence_op_gpu.cu.cc | 35 #define DEFINE_GPU_SPECS(T) \ macro
41 TF_CALL_GPU_NUMBER_TYPES(DEFINE_GPU_SPECS);
42 TF_CALL_bool(DEFINE_GPU_SPECS);
|
| D | mirror_pad_op_gpu.cu.cc | 28 #define DEFINE_GPU_SPECS(T) \ macro
50 TF_CALL_GPU_NUMBER_TYPES(DEFINE_GPU_SPECS);
51 #undef DEFINE_GPU_SPECS
|
| D | compare_and_bitpack_op_gpu.cu.cc | 117 #define DEFINE_GPU_SPECS(T) \ macro
133 TF_CALL_GPU_NUMBER_TYPES(DEFINE_GPU_SPECS)
134 TF_CALL_bool(DEFINE_GPU_SPECS)
|
| D | pooling_ops_3d_gpu.cu.cc | 166 #define DEFINE_GPU_SPECS(T) template struct functor::MaxPool3dGradBackward<T>; macro
167 TF_CALL_GPU_NUMBER_TYPES(DEFINE_GPU_SPECS);
168 #undef DEFINE_GPU_SPECS
|
| D | resize_bilinear_op_gpu.cu.cc | 347 #define DEFINE_GPU_SPECS(T) \ macro
351 TF_CALL_GPU_NUMBER_TYPES_NO_HALF(DEFINE_GPU_SPECS);
353 #undef DEFINE_GPU_SPECS
|
| D | dilation_ops_gpu.cu.cc | 293 #define DEFINE_GPU_SPECS(T) \ macro
298 TF_CALL_GPU_NUMBER_TYPES(DEFINE_GPU_SPECS);
300 #undef DEFINE_GPU_SPECS
|
| D | crop_and_resize_op_gpu.cu.cc | 479 #define DEFINE_GPU_SPECS(T) \ macro
484 TF_CALL_GPU_NUMBER_TYPES(DEFINE_GPU_SPECS);
486 #undef DEFINE_GPU_SPECS
|
| D | bias_op_gpu.cu.cc | 271 #define DEFINE_GPU_SPECS(T) \ macro
275 TF_CALL_GPU_NUMBER_TYPES(DEFINE_GPU_SPECS);
|
| /external/tensorflow/tensorflow/contrib/framework/kernels/ |
| D | zero_initializer_op_gpu.cu.cc | 29 #define DEFINE_GPU_SPECS(T) template struct TensorSetZero<GPUDevice, T>; macro
30 TF_CALL_GPU_NUMBER_TYPES(DEFINE_GPU_SPECS);
31 #undef DEFINE_GPU_SPECS
|
| /external/tensorflow/tensorflow/contrib/rnn/kernels/ |
| D | gru_ops_gpu.cu.cc | 26 #define DEFINE_GPU_SPECS(T) \ macro
30 DEFINE_GPU_SPECS(float);
31 #undef DEFINE_GPU_SPECS
|
| D | lstm_ops_gpu.cu.cc | 402 #define DEFINE_GPU_SPECS(T) \ macro
457 DEFINE_GPU_SPECS(float);
458 DEFINE_GPU_SPECS(Eigen::half);
460 #undef DEFINE_GPU_SPECS
|
| /external/tensorflow/tensorflow/contrib/seq2seq/kernels/ |
| D | beam_search_ops_gpu.cu.cc | 102 #define DEFINE_GPU_SPECS(T) template struct GatherTree<GPUDevice, T>; macro
104 DEFINE_GPU_SPECS(int32);
105 #undef DEFINE_GPU_SPECS
|
| /external/tensorflow/tensorflow/contrib/reduce_slice_ops/kernels/ |
| D | reduce_slice_ops_gpu.cu.cc | 94 #define DEFINE_GPU_SPECS(T) \ macro
97 TF_CALL_REAL_NUMBER_TYPES(DEFINE_GPU_SPECS)
100 #undef DEFINE_GPU_SPECS
|