Searched refs:should_cast (Results 1 – 3 of 3) sorted by relevance
67 ShouldCastFunc should_cast = nullptr; member
201 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()
1691 popts.should_cast = [enable_bfloat16_sendrecv](const Edge* e) { in BuildAndRegisterPartitions()