| /external/tensorflow/tensorflow/core/kernels/ |
| cwise_op_atan2.cc | 21 REGISTER2(BinaryOp, GPU, "Atan2", functor::atan2, float, double);
|
| cwise_op_rint.cc | 21 REGISTER2(UnaryOp, GPU, "Rint", functor::rint, float, double);
|
| cwise_op_acos.cc | 22 REGISTER2(UnaryOp, GPU, "Acos", functor::acos, float, double);
|
| cwise_op_acosh.cc | 28 REGISTER2(UnaryOp, GPU, "Acosh", functor::acosh, float, double);
|
| cwise_op_asin.cc | 22 REGISTER2(UnaryOp, GPU, "Asin", functor::asin, float, double);
|
| cwise_op_asinh.cc | 28 REGISTER2(UnaryOp, GPU, "Asinh", functor::asinh, float, double);
|
| cwise_op_atan.cc | 22 REGISTER2(UnaryOp, GPU, "Atan", functor::atan, float, double);
|
| cwise_op_atanh.cc | 28 REGISTER2(UnaryOp, GPU, "Atanh", functor::atanh, float, double);
|
| cwise_op_digamma.cc | 22 REGISTER3(UnaryOp, GPU, "Digamma", functor::digamma, float, Eigen::half,
|
| cwise_op_erf.cc | 21 REGISTER3(UnaryOp, GPU, "Erf", functor::erf, float, Eigen::half, double);
|
| cwise_op_erfc.cc | 21 REGISTER3(UnaryOp, GPU, "Erfc", functor::erfc, float, Eigen::half, double);
|
| cwise_op_lgamma.cc | 21 REGISTER3(UnaryOp, GPU, "Lgamma", functor::lgamma, float, Eigen::half, double);
|
| cwise_op_not_equal_to_1.cc | 22 REGISTER4(BinaryOp, GPU, "NotEqual", functor::not_equal_to, float, Eigen::half,
|
| cwise_op_tan.cc | 22 REGISTER2(UnaryOp, GPU, "Tan", functor::tan, float, double);
|
| cwise_op_arg.cc | 29 // TODO: Enable GPU support for angle op after resolving
30 // build failures on GPU (See #10643 for context).
32 REGISTER_COMPLEX(GPU, float, complex64);
33 REGISTER_COMPLEX(GPU, double, complex128);
|
| cwise_op_complex.cc | 30 REGISTER_COMPLEX(GPU, float, complex64);
31 REGISTER_COMPLEX(GPU, double, complex128);
|
| cwise_op_imag.cc | 30 REGISTER_COMPLEX(GPU, float, complex64);
31 REGISTER_COMPLEX(GPU, double, complex128);
|
| cwise_op_real.cc | 31 REGISTER_COMPLEX(GPU, float, complex64);
32 REGISTER_COMPLEX(GPU, double, complex128);
|
| cwise_op_tanh.cc | 24 REGISTER3(UnaryOp, GPU, "Tanh", functor::tanh, float, Eigen::half, double);
34 REGISTER3(SimpleBinaryOp, GPU, "TanhGrad", functor::tanh_grad, float,
|
| constant_op.cc | 99 REGISTER_KERNEL(GPU, Eigen::half);
100 REGISTER_KERNEL(GPU, bfloat16);
101 REGISTER_KERNEL(GPU, float);
102 REGISTER_KERNEL(GPU, double);
103 REGISTER_KERNEL(GPU, uint8);
104 REGISTER_KERNEL(GPU, int8);
105 REGISTER_KERNEL(GPU, qint8);
106 REGISTER_KERNEL(GPU, uint16);
107 REGISTER_KERNEL(GPU, int16);
108 REGISTER_KERNEL(GPU, int64)
[all...] |
| cwise_op_add_2.cc | 32 REGISTER4(BinaryOp, GPU, "Add", functor::add, uint8, int64, complex64,
34 REGISTER4(BinaryOp, GPU, "AddV2", functor::add, uint8, int64, complex64,
|
| cwise_op_rsqrt.cc | 23 REGISTER3(UnaryOp, GPU, "Rsqrt", functor::rsqrt, float, Eigen::half, double);
32 REGISTER3(SimpleBinaryOp, GPU, "RsqrtGrad", functor::rsqrt_grad, float,
|
| cwise_op_sigmoid.cc | 23 REGISTER3(UnaryOp, GPU, "Sigmoid", functor::sigmoid, float, Eigen::half,
33 REGISTER3(SimpleBinaryOp, GPU, "SigmoidGrad", functor::sigmoid_grad, float,
|
| cwise_op_sqrt.cc | 23 REGISTER3(UnaryOp, GPU, "Sqrt", functor::sqrt, float, Eigen::half, double);
33 REGISTER3(SimpleBinaryOp, GPU, "SqrtGrad", functor::sqrt_grad, float,
|
| /external/llvm/lib/Target/AMDGPU/ |
| AMDGPUSubtarget.cpp | 38 StringRef GPU, StringRef FS) {
53 ParseSubtargetFeatures(GPU, FullFS);
70 AMDGPUSubtarget::AMDGPUSubtarget(const Triple &TT, StringRef GPU, StringRef FS,
72 : AMDGPUGenSubtargetInfo(TT, GPU, FS),
119 InstrItins(getInstrItineraryForCPU(GPU)) {
120 initializeSubtargetDependencies(TT, GPU, FS);
181 R600Subtarget::R600Subtarget(const Triple &TT, StringRef GPU, StringRef FS,
183 AMDGPUSubtarget(TT, GPU, FS, TM),
188 SISubtarget::SISubtarget(const Triple &TT, StringRef GPU, StringRef FS,
190 AMDGPUSubtarget(TT, GPU, FS, TM)
[all...] |
