| /external/tensorflow/tensorflow/core/kernels/ |
| cwise_op_gpu_floor_mod.cu.cc | 21 namespace functor { namespace in namespace:tensorflow
23 } // namespace functor
|
| cwise_op_gpu_mod.cu.cc | 21 namespace functor { namespace in namespace:tensorflow
23 } // namespace functor
|
| cwise_op_gpu_acos.cu.cc | 21 namespace functor { namespace in namespace:tensorflow
23 } // namespace functor
|
| cwise_op_gpu_acosh.cu.cc | 22 namespace functor { namespace in namespace:tensorflow
24 } // namespace functor
|
| cwise_op_gpu_asin.cu.cc | 21 namespace functor { namespace in namespace:tensorflow
23 } // namespace functor
|
| cwise_op_gpu_asinh.cu.cc | 22 namespace functor { namespace in namespace:tensorflow
24 } // namespace functor
|
| cwise_op_gpu_atan.cu.cc | 21 namespace functor { namespace in namespace:tensorflow
23 } // namespace functor
|
| cwise_op_gpu_atan2.cu.cc | 21 namespace functor { namespace in namespace:tensorflow
23 } // namespace functor
|
| cwise_op_gpu_atanh.cu.cc | 22 namespace functor { namespace in namespace:tensorflow
24 } // namespace functor
|
| cwise_op_gpu_complex.cu.cc | 21 namespace functor { namespace in namespace:tensorflow
23 } // namespace functor
|
| cwise_op_gpu_cosh.cu.cc | 21 namespace functor { namespace in namespace:tensorflow
23 } // namespace functor
|
| cwise_op_gpu_rint.cu.cc | 21 namespace functor { namespace in namespace:tensorflow
23 } // namespace functor
|
| cwise_op_gpu_sinh.cu.cc | 21 namespace functor { namespace in namespace:tensorflow
23 } // namespace functor
|
| cwise_op_gpu_tan.cu.cc | 21 namespace functor { namespace in namespace:tensorflow
23 } // namespace functor
|
| adjust_contrast_op_gpu.cu.cc | 29 template struct functor::AdjustContrastv2<GPUDevice>;
32 template struct functor::AdjustContrast<GPUDevice, uint8>;
33 template struct functor::AdjustContrast<GPUDevice, int8>;
34 template struct functor::AdjustContrast<GPUDevice, int16>;
35 template struct functor::AdjustContrast<GPUDevice, int32>;
36 template struct functor::AdjustContrast<GPUDevice, int64>;
37 template struct functor::AdjustContrast<GPUDevice, float>;
38 template struct functor::AdjustContrast<GPUDevice, double>;
|
| strided_slice_op_gpu.cu.cc | 31 template struct functor::StridedSlice<GPUDevice, T, 1>; \
32 template struct functor::StridedSlice<GPUDevice, T, 2>; \
33 template struct functor::StridedSlice<GPUDevice, T, 3>; \
34 template struct functor::StridedSlice<GPUDevice, T, 4>; \
35 template struct functor::StridedSlice<GPUDevice, T, 5>; \
36 template struct functor::StridedSlice<GPUDevice, T, 6>; \
37 template struct functor::StridedSlice<GPUDevice, T, 7>; \
38 template struct functor::StridedSliceGrad<GPUDevice, T, 1>; \
39 template struct functor::StridedSliceGrad<GPUDevice, T, 2>; \
40 template struct functor::StridedSliceGrad<GPUDevice, T, 3>;
[all...] |
| mirror_pad_op_gpu.cu.cc | 29 template struct functor::MirrorPad<GpuDevice, T, int32, 1>; \
30 template struct functor::MirrorPad<GpuDevice, T, int32, 2>; \
31 template struct functor::MirrorPad<GpuDevice, T, int32, 3>; \
32 template struct functor::MirrorPad<GpuDevice, T, int32, 4>; \
33 template struct functor::MirrorPad<GpuDevice, T, int32, 5>; \
34 template struct functor::MirrorPad<GpuDevice, T, int64, 1>; \
35 template struct functor::MirrorPad<GpuDevice, T, int64, 2>; \
36 template struct functor::MirrorPad<GpuDevice, T, int64, 3>; \
37 template struct functor::MirrorPad<GpuDevice, T, int64, 4>; \
38 template struct functor::MirrorPad<GpuDevice, T, int64, 5>;
[all...] |
| cwise_op_acos.cc | 19 REGISTER2(UnaryOp, CPU, "Acos", functor::acos, float, double);
22 REGISTER2(UnaryOp, GPU, "Acos", functor::acos, float, double);
26 REGISTER2(UnaryOp, SYCL, "Acos", functor::acos, float, double);
|
| cwise_op_acosh.cc | 20 REGISTER4(UnaryOp, CPU, "Acosh", functor::acosh, float, double, complex64,
24 REGISTER2(UnaryOp, SYCL, "Acosh", functor::acosh, float, double);
28 REGISTER2(UnaryOp, GPU, "Acosh", functor::acosh, float, double);
|
| cwise_op_asin.cc | 19 REGISTER2(UnaryOp, CPU, "Asin", functor::asin, float, double);
22 REGISTER2(UnaryOp, GPU, "Asin", functor::asin, float, double);
26 REGISTER2(UnaryOp, SYCL, "Asin", functor::asin, float, double);
|
| cwise_op_asinh.cc | 20 REGISTER4(UnaryOp, CPU, "Asinh", functor::asinh, float, double, complex64,
24 REGISTER2(UnaryOp, SYCL, "Asinh", functor::asinh, float, double);
28 REGISTER2(UnaryOp, GPU, "Asinh", functor::asinh, float, double);
|
| cwise_op_atan.cc | 19 REGISTER2(UnaryOp, CPU, "Atan", functor::atan, float, double);
22 REGISTER2(UnaryOp, GPU, "Atan", functor::atan, float, double);
26 REGISTER2(UnaryOp, SYCL, "Atan", functor::atan, float, double);
|
| cwise_op_atanh.cc | 20 REGISTER4(UnaryOp, CPU, "Atanh", functor::atanh, float, double, complex64,
24 REGISTER2(UnaryOp, SYCL, "Atanh", functor::atanh, float, double);
28 REGISTER2(UnaryOp, GPU, "Atanh", functor::atanh, float, double);
|
| cwise_op_tan.cc | 19 REGISTER2(UnaryOp, CPU, "Tan", functor::tan, float, double);
22 REGISTER2(UnaryOp, GPU, "Tan", functor::tan, float, double);
26 REGISTER2(UnaryOp, SYCL, "Tan", functor::tan, float, double);
|
| cwise_op_atan2.cc | 19 REGISTER2(BinaryOp, CPU, "Atan2", functor::atan2, float, double);
21 REGISTER2(BinaryOp, GPU, "Atan2", functor::atan2, float, double);
|
