Home
last modified time | relevance | path

Searched refs:GetGpuCastFromUint8 (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcast_op_impl_uint8.cc30 CastFunctorType GetGpuCastFromUint8(DataType dst_dtype) { in GetGpuCastFromUint8() function
Dcast_op_impl.h104 CastFunctorType GetGpuCastFromUint8(DataType dst_dtype);
Dcast_op.cc186 work_ = GetGpuCastFromUint8(dst_dtype_); in Prepare()