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_double.cc23 CastFunctorType GetCpuCastFromDouble(DataType dst_dtype) { in GetCpuCastFromDouble()
29 CastFunctorType GetGpuCastFromDouble(DataType dst_dtype) { in GetGpuCastFromDouble()
37 CastFunctorType GetSyclCastFromDouble(DataType dst_dtype) { in GetSyclCastFromDouble()
Dcast_op_impl_int16.cc23 CastFunctorType GetCpuCastFromInt16(DataType dst_dtype) { in GetCpuCastFromInt16()
29 CastFunctorType GetGpuCastFromInt16(DataType dst_dtype) { in GetGpuCastFromInt16()
37 CastFunctorType GetSyclCastFromInt16(DataType dst_dtype) { in GetSyclCastFromInt16()
Dcast_op_impl_float.cc25 CastFunctorType GetCpuCastFromFloat(DataType dst_dtype) { in GetCpuCastFromFloat()
31 CastFunctorType GetGpuCastFromFloat(DataType dst_dtype) { in GetGpuCastFromFloat()
39 CastFunctorType GetSyclCastFromFloat(DataType dst_dtype) { in GetSyclCastFromFloat()
Dcast_op_impl_uint64.cc23 CastFunctorType GetCpuCastFromUint64(DataType dst_dtype) { in GetCpuCastFromUint64()
29 CastFunctorType GetGpuCastFromUint64(DataType dst_dtype) { in GetGpuCastFromUint64()
37 CastFunctorType GetSyclCastFromUint64(DataType dst_dtype) { in GetSyclCastFromUint64()
Dcast_op_impl_int64.cc23 CastFunctorType GetCpuCastFromInt64(DataType dst_dtype) { in GetCpuCastFromInt64()
29 CastFunctorType GetGpuCastFromInt64(DataType dst_dtype) { in GetGpuCastFromInt64()
37 CastFunctorType GetSyclCastFromInt64(DataType dst_dtype) { in GetSyclCastFromInt64()
Dcast_op_impl_uint32.cc23 CastFunctorType GetCpuCastFromUint32(DataType dst_dtype) { in GetCpuCastFromUint32()
29 CastFunctorType GetGpuCastFromUint32(DataType dst_dtype) { in GetGpuCastFromUint32()
37 CastFunctorType GetSyclCastFromUint32(DataType dst_dtype) { in GetSyclCastFromUint32()
Dcast_op_impl_int32.cc23 CastFunctorType GetCpuCastFromInt32(DataType dst_dtype) { in GetCpuCastFromInt32()
29 CastFunctorType GetGpuCastFromInt32(DataType dst_dtype) { in GetGpuCastFromInt32()
37 CastFunctorType GetSyclCastFromInt32(DataType dst_dtype) { in GetSyclCastFromInt32()
Dcast_op_impl_uint8.cc23 CastFunctorType GetCpuCastFromUint8(DataType dst_dtype) { in GetCpuCastFromUint8()
29 CastFunctorType GetGpuCastFromUint8(DataType dst_dtype) { in GetGpuCastFromUint8()
37 CastFunctorType GetSyclCastFromUint8(DataType dst_dtype) { in GetSyclCastFromUint8()
Dcast_op_impl_uint16.cc23 CastFunctorType GetCpuCastFromUint16(DataType dst_dtype) { in GetCpuCastFromUint16()
29 CastFunctorType GetGpuCastFromUint16(DataType dst_dtype) { in GetGpuCastFromUint16()
37 CastFunctorType GetSyclCastFromUint16(DataType dst_dtype) { in GetSyclCastFromUint16()
Dcast_op_impl_bool.cc23 CastFunctorType GetCpuCastFromBool(DataType dst_dtype) { in GetCpuCastFromBool()
29 CastFunctorType GetGpuCastFromBool(DataType dst_dtype) { in GetGpuCastFromBool()
37 CastFunctorType GetSyclCastFromBool(DataType dst_dtype) { in GetSyclCastFromBool()
Dcast_op_impl_int8.cc23 CastFunctorType GetCpuCastFromInt8(DataType dst_dtype) { in GetCpuCastFromInt8()
29 CastFunctorType GetGpuCastFromInt8(DataType dst_dtype) { in GetGpuCastFromInt8()
37 CastFunctorType GetSyclCastFromInt8(DataType dst_dtype) { in GetSyclCastFromInt8()
Dcast_op_impl_half.cc23 CastFunctorType GetCpuCastFromHalf(DataType dst_dtype) { in GetCpuCastFromHalf()
29 CastFunctorType GetGpuCastFromHalf(DataType dst_dtype) { in GetGpuCastFromHalf()
Dcast_op_impl_complex128.cc23 CastFunctorType GetCpuCastFromComplex128(DataType dst_dtype) { in GetCpuCastFromComplex128()
29 CastFunctorType GetGpuCastFromComplex128(DataType dst_dtype) { in GetGpuCastFromComplex128()
Dcast_op_impl_complex64.cc23 CastFunctorType GetCpuCastFromComplex64(DataType dst_dtype) { in GetCpuCastFromComplex64()
29 CastFunctorType GetGpuCastFromComplex64(DataType dst_dtype) { in GetGpuCastFromComplex64()
Dcast_op_impl_bfloat.cc25 CastFunctorType GetCpuCastFromBfloat(DataType dst_dtype) { in GetCpuCastFromBfloat()
31 CastFunctorType GetGpuCastFromBfloat(DataType dst_dtype) { in GetGpuCastFromBfloat()
/external/tensorflow/tensorflow/python/kernel_tests/
Dcast_op_test.py126 def _compare(self, x, dst_dtype, expected, use_gpu=False): argument
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client_test.py541 def _ConvertAndTest(template, src_dtype, dst_dtype): argument
568 def _ConvertAndTest(template, src_dtype, dst_dtype, dst_etype): argument