Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dcast_op_impl_bfloat.cc32 CastFunctorType GetGpuCastFromBfloat(DataType dst_dtype) { in GetGpuCastFromBfloat() function
Dcast_op_impl.h130 CastFunctorType GetGpuCastFromBfloat(DataType dst_dtype);
Dcast_op.cc212 work_ = GetGpuCastFromBfloat(dst_dtype_); in Prepare()