Searched refs:CastFunctor (Results 1 – 5 of 5) 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 | 35 functor::CastFunctor<GPUDevice, float, bfloat16> func; in GetGpuCastFromBfloat()
|
| D | cast_op_gpu.cu.cc | 32 #define DEFINE(O, I) template struct CastFunctor<GPUDevice, O, I>
|
| D | self_adjoint_eig_v2_op_gpu.cc | 143 functor::CastFunctor<GPUDevice, Scalar, RealScalar> cast; in ComputeAsync()
|
| D | cast_op_impl.h | 63 functor::CastFunctor<DEVICE, OUT, IN> func; \
|