Home
last modified time | relevance | path

Searched defs:dst_dtype (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcast_op_impl_int8.cc23 CastFunctorType GetCpuCastFromInt8(DataType dst_dtype) { in GetCpuCastFromInt8()
30 CastFunctorType GetGpuCastFromInt8(DataType dst_dtype) { in GetGpuCastFromInt8()
Dcast_op_impl_double.cc23 CastFunctorType GetCpuCastFromDouble(DataType dst_dtype) { in GetCpuCastFromDouble()
30 CastFunctorType GetGpuCastFromDouble(DataType dst_dtype) { in GetGpuCastFromDouble()
Dcast_op_impl_uint32.cc23 CastFunctorType GetCpuCastFromUint32(DataType dst_dtype) { in GetCpuCastFromUint32()
30 CastFunctorType GetGpuCastFromUint32(DataType dst_dtype) { in GetGpuCastFromUint32()
Dcast_op_impl_int32.cc23 CastFunctorType GetCpuCastFromInt32(DataType dst_dtype) { in GetCpuCastFromInt32()
30 CastFunctorType GetGpuCastFromInt32(DataType dst_dtype) { in GetGpuCastFromInt32()
Dcast_op_impl_half.cc23 CastFunctorType GetCpuCastFromHalf(DataType dst_dtype) { in GetCpuCastFromHalf()
30 CastFunctorType GetGpuCastFromHalf(DataType dst_dtype) { in GetGpuCastFromHalf()
Dcast_op_impl_int16.cc23 CastFunctorType GetCpuCastFromInt16(DataType dst_dtype) { in GetCpuCastFromInt16()
30 CastFunctorType GetGpuCastFromInt16(DataType dst_dtype) { in GetGpuCastFromInt16()
Dcast_op_impl_uint16.cc23 CastFunctorType GetCpuCastFromUint16(DataType dst_dtype) { in GetCpuCastFromUint16()
30 CastFunctorType GetGpuCastFromUint16(DataType dst_dtype) { in GetGpuCastFromUint16()
Dcast_op_impl_int64.cc23 CastFunctorType GetCpuCastFromInt64(DataType dst_dtype) { in GetCpuCastFromInt64()
30 CastFunctorType GetGpuCastFromInt64(DataType dst_dtype) { in GetGpuCastFromInt64()
Dcast_op_impl_complex64.cc23 CastFunctorType GetCpuCastFromComplex64(DataType dst_dtype) { in GetCpuCastFromComplex64()
30 CastFunctorType GetGpuCastFromComplex64(DataType dst_dtype) { in GetGpuCastFromComplex64()
Dcast_op_impl_bool.cc23 CastFunctorType GetCpuCastFromBool(DataType dst_dtype) { in GetCpuCastFromBool()
30 CastFunctorType GetGpuCastFromBool(DataType dst_dtype) { in GetGpuCastFromBool()
Dcast_op_impl_uint8.cc23 CastFunctorType GetCpuCastFromUint8(DataType dst_dtype) { in GetCpuCastFromUint8()
30 CastFunctorType GetGpuCastFromUint8(DataType dst_dtype) { in GetGpuCastFromUint8()
Dcast_op_impl_float.cc25 CastFunctorType GetCpuCastFromFloat(DataType dst_dtype) { in GetCpuCastFromFloat()
32 CastFunctorType GetGpuCastFromFloat(DataType dst_dtype) { in GetGpuCastFromFloat()
Dcast_op_impl_uint64.cc23 CastFunctorType GetCpuCastFromUint64(DataType dst_dtype) { in GetCpuCastFromUint64()
30 CastFunctorType GetGpuCastFromUint64(DataType dst_dtype) { in GetGpuCastFromUint64()
Dcast_op_impl_complex128.cc23 CastFunctorType GetCpuCastFromComplex128(DataType dst_dtype) { in GetCpuCastFromComplex128()
30 CastFunctorType GetGpuCastFromComplex128(DataType dst_dtype) { in GetGpuCastFromComplex128()
Dcast_op_impl_bfloat.cc25 CastFunctorType GetCpuCastFromBfloat(DataType dst_dtype) { in GetCpuCastFromBfloat()
32 CastFunctorType GetGpuCastFromBfloat(DataType dst_dtype) { in GetGpuCastFromBfloat()
/external/tensorflow/tensorflow/python/kernel_tests/
Dcast_op_test.py122 def _compare(self, x, dst_dtype, expected, use_gpu=False): argument
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client_test.py592 def testConvertElementType(self, src_dtype, dst_dtype): argument
622 def testBitcastConvertType(self, src_dtype, dst_dtype): argument