Searched refs:num_replicated_inputs (Results 1 – 2 of 2) sorted by relevance
355 int32_t num_replicated_inputs = replicated_inputs.size() * n; in parse() local357 … auto attr = parser.getBuilder().getDenseI32ArrayAttr({num_replicated_inputs, num_packed_inputs}); in parse()388 const int32_t num_replicated_inputs = in print() local390 const int32_t num_replicated_block_args = num_replicated_inputs / n; in print()464 int32_t num_replicated_inputs = replicated_inputs.size() * n; in BuildReplicateOp() local467 builder->getDenseI32ArrayAttr({num_replicated_inputs, num_packed_inputs}); in BuildReplicateOp()505 const int32_t num_replicated_inputs = in verify() local510 if (num_replicated_inputs % n != 0) in verify()512 << "expects number of replicated inputs (" << num_replicated_inputs in verify()515 const int32_t num_replicated_block_args = num_replicated_inputs / n; in verify()[all …]
906 int num_replicated_inputs = num_per_replica_inputs + num_distributed_vars; in RemoveUnusedXlaInput() local909 int arg_index = num_replicated_inputs + i; in RemoveUnusedXlaInput()918 int arg_index = num_replicated_inputs + broadcast_input_types.size() + i; in RemoveUnusedXlaInput()927 int arg_index = num_replicated_inputs + broadcast_input_types.size() + in RemoveUnusedXlaInput()2622 const int num_replicated_inputs = distributed_var_start_index; in BuildTPUReplicateOps() local2623 const int num_distributed_vars = pos - num_replicated_inputs; in BuildTPUReplicateOps()2634 << " num_replicated_inputs: " << num_replicated_inputs in BuildTPUReplicateOps()2641 in_edges.size() - (num_replicated_inputs + num_distributed_vars + in BuildTPUReplicateOps()2645 const int num_inputs = num_replicated_inputs * num_replicas + in BuildTPUReplicateOps()2660 DataTypeVector replicated_input_types(num_replicated_inputs * num_replicas + in BuildTPUReplicateOps()[all …]