Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dcast_op_impl_int16.cc30 CastFunctorType GetGpuCastFromInt16(DataType dst_dtype) { in GetGpuCastFromInt16() function
Dcast_op_impl.h114 CastFunctorType GetGpuCastFromInt16(DataType dst_dtype);
Dcast_op.cc196 work_ = GetGpuCastFromInt16(dst_dtype_); in Prepare()