Home
last modified time | relevance | path

Searched refs:CAST_CASE (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcast_op_impl_bool.cc24 CURRY_TYPES3(CAST_CASE, CPUDevice, bool); in GetCpuCastFromBool()
31 CURRY_TYPES3_NO_BF16(CAST_CASE, GPUDevice, bool); in GetGpuCastFromBool()
Dcast_op_impl_float.cc26 CURRY_TYPES3(CAST_CASE, CPUDevice, float); in GetCpuCastFromFloat()
33 CURRY_TYPES3(CAST_CASE, GPUDevice, float); in GetGpuCastFromFloat()
Dcast_op_impl_double.cc24 CURRY_TYPES3(CAST_CASE, CPUDevice, double); in GetCpuCastFromDouble()
31 CURRY_TYPES3_NO_BF16(CAST_CASE, GPUDevice, double); in GetGpuCastFromDouble()
Dcast_op_impl_uint64.cc24 CURRY_TYPES3(CAST_CASE, CPUDevice, uint64); in GetCpuCastFromUint64()
31 CURRY_TYPES3_NO_BF16(CAST_CASE, GPUDevice, uint64); in GetGpuCastFromUint64()
Dcast_op_impl_int32.cc24 CURRY_TYPES3(CAST_CASE, CPUDevice, int32); in GetCpuCastFromInt32()
31 CURRY_TYPES3_NO_BF16(CAST_CASE, GPUDevice, int32); in GetGpuCastFromInt32()
Dcast_op_impl_int64.cc24 CURRY_TYPES3(CAST_CASE, CPUDevice, int64); in GetCpuCastFromInt64()
31 CURRY_TYPES3_NO_BF16(CAST_CASE, GPUDevice, int64); in GetGpuCastFromInt64()
Dcast_op_impl_int16.cc24 CURRY_TYPES3(CAST_CASE, CPUDevice, int16); in GetCpuCastFromInt16()
31 CURRY_TYPES3_NO_BF16(CAST_CASE, GPUDevice, int16); in GetGpuCastFromInt16()
Dcast_op_impl_int8.cc24 CURRY_TYPES3(CAST_CASE, CPUDevice, int8); in GetCpuCastFromInt8()
31 CURRY_TYPES3_NO_BF16(CAST_CASE, GPUDevice, int8); in GetGpuCastFromInt8()
Dcast_op_impl_uint32.cc24 CURRY_TYPES3(CAST_CASE, CPUDevice, uint32); in GetCpuCastFromUint32()
31 CURRY_TYPES3_NO_BF16(CAST_CASE, GPUDevice, uint32); in GetGpuCastFromUint32()
Dcast_op_impl_uint16.cc24 CURRY_TYPES3(CAST_CASE, CPUDevice, uint16); in GetCpuCastFromUint16()
31 CURRY_TYPES3_NO_BF16(CAST_CASE, GPUDevice, uint16); in GetGpuCastFromUint16()
Dcast_op_impl_uint8.cc24 CURRY_TYPES3(CAST_CASE, CPUDevice, uint8); in GetCpuCastFromUint8()
31 CURRY_TYPES3_NO_BF16(CAST_CASE, GPUDevice, uint8); in GetGpuCastFromUint8()
Dcast_op_impl_half.cc24 CURRY_TYPES3(CAST_CASE, CPUDevice, Eigen::half); in GetCpuCastFromHalf()
31 CURRY_TYPES3_NO_BF16(CAST_CASE, GPUDevice, Eigen::half); in GetGpuCastFromHalf()
Dcast_op_impl_complex128.cc24 CURRY_TYPES3(CAST_CASE, CPUDevice, std::complex<double>); in GetCpuCastFromComplex128()
31 CURRY_TYPES3_NO_BF16(CAST_CASE, GPUDevice, std::complex<double>); in GetGpuCastFromComplex128()
Dcast_op_impl_complex64.cc24 CURRY_TYPES3(CAST_CASE, CPUDevice, std::complex<float>); in GetCpuCastFromComplex64()
31 CURRY_TYPES3_NO_BF16(CAST_CASE, GPUDevice, std::complex<float>); in GetGpuCastFromComplex64()
Dcast_op_impl_bfloat.cc26 CURRY_TYPES3(CAST_CASE, CPUDevice, bfloat16); in GetCpuCastFromBfloat()
Dcast_op_impl.h56 #define CAST_CASE(DEVICE, IN, OUT) \ macro
Dcast_op.cc220 #undef CAST_CASE