Searched refs:num_replicated_inputs (Results 1 – 2 of 2) sorted by relevance
344 int32_t num_replicated_inputs = replicated_inputs.size() * n; in ParseReplicateOp() local348 {num_replicated_inputs, num_packed_inputs}); in ParseReplicateOp()381 const int32_t num_replicated_inputs = in Print() local383 const int32_t num_replicated_block_args = num_replicated_inputs / n; in Print()451 const int32_t num_replicated_inputs = in Verify() local456 if (num_replicated_inputs % n != 0) in Verify()458 << "expects number of replicated inputs (" << num_replicated_inputs in Verify()461 const int32_t num_replicated_block_args = num_replicated_inputs / n; in Verify()466 << num_replicated_inputs << ") / 'n' (" << n in Verify()487 num_replicated_inputs; in Verify()[all …]
937 int num_replicated_inputs = num_per_replica_inputs + num_distributed_vars; in RemoveUnusedXlaInput() local940 int arg_index = num_replicated_inputs + i; in RemoveUnusedXlaInput()949 int arg_index = num_replicated_inputs + broadcast_input_types.size() + i; in RemoveUnusedXlaInput()958 int arg_index = num_replicated_inputs + broadcast_input_types.size() + in RemoveUnusedXlaInput()2670 const int num_replicated_inputs = distributed_var_start_index; in BuildTPUReplicateOps() local2671 const int num_distributed_vars = pos - num_replicated_inputs; in BuildTPUReplicateOps()2682 << " num_replicated_inputs: " << num_replicated_inputs in BuildTPUReplicateOps()2689 in_edges.size() - (num_replicated_inputs + num_distributed_vars + in BuildTPUReplicateOps()2693 const int num_inputs = num_replicated_inputs * num_replicas + in BuildTPUReplicateOps()2708 DataTypeVector replicated_input_types(num_replicated_inputs * num_replicas + in BuildTPUReplicateOps()[all …]