Home
last modified time | relevance | path

Searched refs:num_replicated_block_args (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_device.cc307 const int32_t num_replicated_block_args = replicated_inputs.size(); in SetReplicateOpOperands() local
314 packed_input, region_arg_types[idx + num_replicated_block_args], in SetReplicateOpOperands()
390 const int32_t num_replicated_block_args = num_replicated_inputs / n; in print() local
397 if (block_arg_num < num_replicated_block_args) { in print()
405 block_arg_num - num_replicated_block_args)); in print()
515 const int32_t num_replicated_block_args = num_replicated_inputs / n; in verify() local
516 if (num_replicated_block_args + num_packed_inputs != block.getNumArguments()) in verify()
536 if (block_arg.getArgNumber() < num_replicated_block_args) { in verify()
540 const int32_t idx = block_arg.getArgNumber() - num_replicated_block_args + in verify()