Searched refs:num_broadcast_inputs (Results 1 – 1 of 1) sorted by relevance
2688 const int num_broadcast_inputs = in BuildTPUReplicateOps() local2691 TF_RET_CHECK(num_broadcast_inputs >= 0); in BuildTPUReplicateOps()2694 num_distributed_vars + num_broadcast_inputs + in BuildTPUReplicateOps()2742 DataTypeVector broadcast_input_types(num_broadcast_inputs); in BuildTPUReplicateOps()2743 for (int i = 0; i < num_broadcast_inputs; ++i) { in BuildTPUReplicateOps()2754 num_broadcast_inputs + i; in BuildTPUReplicateOps()2756 num_broadcast_inputs + i]; in BuildTPUReplicateOps()2763 num_broadcast_inputs + num_variables + i; in BuildTPUReplicateOps()2765 num_broadcast_inputs + num_variables + i]; in BuildTPUReplicateOps()