Searched refs:CastFunctor (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cast_op.h | 32 struct CastFunctor<Device, OUT_TYPE, IN_OUT> { \
46 template struct CastFunctor<DEVICE, OUT_TYPE, IN_OUT>;
50 struct CastFunctor<DEVICE, OUT_TYPE, IN_OUT> { \
77 struct CastFunctor<devname, OUT_TYPE, IN_OUT> { \
146 struct CastFunctor { struct
|
| D | cast_op_impl_bfloat.cc | 36 functor::CastFunctor<GPUDevice, float, bfloat16> func; in GetGpuCastFromBfloat()
|
| D | cast_op_gpu.cu.cc | 33 #define DEFINE(O, I) template struct CastFunctor<GPUDevice, O, I>
|
| D | cast_op_impl.h | 60 functor::CastFunctor<DEVICE, OUT, IN> func; \
|
| D | depthwise_conv_grad_op.cc | 1191 functor::CastFunctor<Device, float, Eigen::half> cast; in Compute()
1214 functor::CastFunctor<Device, Eigen::half, float> cast; in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | self_adjoint_eig_v2_op_gpu.cc | 143 functor::CastFunctor<GPUDevice, Scalar, RealScalar> cast; in ComputeAsync()
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | resize_bilinear_op.cc | 131 functor::CastFunctor<GPUDevice, T, float> cast; in operator ()()
|