Searched refs:enable_bfloat16_sendrecv (Results 1 – 3 of 3) sorted by relevance
43 name: "enable_bfloat16_sendrecv"
1597 const bool enable_bfloat16_sendrecv = in BuildAndRegisterPartitions() local1598 session_opts_.config.graph_options().enable_bfloat16_sendrecv(); in BuildAndRegisterPartitions()1599 popts.should_cast = [enable_bfloat16_sendrecv](const Edge* e) { in BuildAndRegisterPartitions()1604 if (enable_bfloat16_sendrecv && dtype == DT_FLOAT) { in BuildAndRegisterPartitions()
260 bool enable_bfloat16_sendrecv = 7; field