Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/graph/
Dgraph_partition.h67 ShouldCastFunc should_cast = nullptr; member
Dgraph_partition.cc201 const DataType cast_dtype = opts.should_cast ? opts.should_cast(edge) : dtype; in AddSend()
264 if (opts.should_cast && !NeedSameDeviceSendRecv(edge, g_info)) { in AddRecv()
265 cast_dtype = opts.should_cast(edge); in AddRecv()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dmaster_session.cc1691 popts.should_cast = [enable_bfloat16_sendrecv](const Edge* e) { in BuildAndRegisterPartitions()