Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_device.cc654 unsigned num_replicated_args = GetNumReplicatedBlockArguments(); in GetOperandsForBlockArgument() local
660 if (arg_number < num_replicated_args) in GetOperandsForBlockArgument()
663 operands = operands.drop_front(num_replicated_args * num_replicas); in GetOperandsForBlockArgument()
664 arg_number -= num_replicated_args; in GetOperandsForBlockArgument()