| /external/tensorflow/tensorflow/core/kernels/ |
| D | shape_ops.cc | 38 #define REGISTER_GPU_KERNEL(type) \ macro
127 #define REGISTER_GPU_KERNEL(type) \ macro
205 #define REGISTER_GPU_KERNEL(type) \ macro
274 #define REGISTER_GPU_KERNEL(type) \ macro
361 #define REGISTER_GPU_KERNEL(type) \ macro
434 #define REGISTER_GPU_KERNEL(type) \ macro
|
| D | control_flow_ops.cc | 243 #define REGISTER_GPU_KERNEL(type) \ macro
307 #define REGISTER_GPU_KERNEL(type) \ macro
364 #define REGISTER_GPU_KERNEL(type) \ macro
419 #define REGISTER_GPU_KERNEL(type) \ macro
|
| D | reshape_op.cc | 32 #define REGISTER_GPU_KERNEL(type) \ macro
|
| D | split_lib_gpu.h | 48 #define REGISTER_GPU_KERNEL(T) \ macro
|
| D | identity_op.cc | 56 #define REGISTER_GPU_KERNEL(type) \ macro
|
| D | batch_norm_op.cc | 196 #define REGISTER_GPU_KERNEL(T) \ macro
244 #define REGISTER_GPU_KERNEL(T) \ macro
|
| D | l2loss_op_gpu.cu.cc | 63 #define REGISTER_GPU_KERNEL(T) \ macro
|
| D | data_format_ops.cc | 304 #define REGISTER_GPU_KERNEL(T) \ macro
312 #define REGISTER_GPU_KERNEL(T) \ macro
|
| D | cross_op.cc | 104 #define REGISTER_GPU_KERNEL(type) \ macro
|
| D | split_lib_gpu.cu.cc | 244 #define REGISTER_GPU_KERNEL(T) template struct SplitOpGPULaunch<T>; macro
250 #define REGISTER_GPU_KERNEL(T) \ macro
|
| D | cwise_op_clip.cc | 188 #define REGISTER_GPU_KERNEL(type) \ macro
|
| D | sequence_ops.cc | 129 #define REGISTER_GPU_KERNEL(T) REGISTER_KERNEL(DEVICE_GPU, GPUDevice, T) macro
|
| D | bias_op.cc | 286 #define REGISTER_GPU_KERNEL(type) \ macro
536 #define REGISTER_GPU_KERNEL(type) \ macro
|
| D | pad_op.cc | 323 #define REGISTER_GPU_KERNEL(T) \ macro
|
| D | quantize_and_dequantize_op.cc | 468 #define REGISTER_GPU_KERNEL(T) \ macro
|
| D | depthwise_conv_op.cc | 525 #define REGISTER_GPU_KERNEL(T) \ macro
|
| D | depthwise_conv_grad_op.cc | 727 #define REGISTER_GPU_KERNEL(T) \ macro
1266 #define REGISTER_GPU_KERNEL(T) \ macro
|
| D | conv_grad_ops_3d.cc | 1956 #define REGISTER_GPU_KERNEL(T) \ macro
|
| /external/tensorflow/tensorflow/core/kernels/rnn/ |
| D | gru_ops.cc | 408 #define REGISTER_GPU_KERNEL(T) \ macro
441 #define REGISTER_GPU_KERNEL(T) \ macro
|
| D | lstm_ops.cc | 506 #define REGISTER_GPU_KERNEL(T) \ macro
769 #define REGISTER_GPU_KERNEL(T) \ macro
1111 #define REGISTER_GPU_KERNEL(T) \ macro
1457 #define REGISTER_GPU_KERNEL(T) \ macro
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | mirror_pad_op.cc | 232 #define REGISTER_GPU_KERNEL(T) \ macro
439 #define REGISTER_GPU_KERNEL(T) \ macro
|
| D | adjust_contrast_op.cc | 136 #define REGISTER_GPU_KERNEL(T) \ macro
|
| /external/tensorflow/tensorflow/core/kernels/mlir_generated/ |
| D | base_gpu_op.h | 31 #define REGISTER_GPU_KERNEL(tf_op, input_type, output_type) \ macro
|