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.cc305 const int32_t num_replicated_block_args = replicated_inputs.size(); in SetReplicateOpOperands() local
312 packed_input, region_arg_types[idx + num_replicated_block_args], in SetReplicateOpOperands()
383 const int32_t num_replicated_block_args = num_replicated_inputs / n; in Print() local
390 if (block_arg_num < num_replicated_block_args) { in Print()
398 block_arg_num - num_replicated_block_args)); in Print()
461 const int32_t num_replicated_block_args = num_replicated_inputs / n; in Verify() local
462 if (num_replicated_block_args + num_packed_inputs != block.getNumArguments()) in Verify()
482 if (block_arg.getArgNumber() < num_replicated_block_args) { in Verify()
486 const int32_t idx = block_arg.getArgNumber() - num_replicated_block_args + in Verify()