Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dcast_op_impl_uint16.cc30 CastFunctorType GetGpuCastFromUint16(DataType dst_dtype) { in GetGpuCastFromUint16() function
Dcast_op_impl.h106 CastFunctorType GetGpuCastFromUint16(DataType dst_dtype);
Dcast_op.cc188 work_ = GetGpuCastFromUint16(dst_dtype_); in Prepare()