Searched defs:replicated_rank (Results 1 – 1 of 1) sorted by relevance
218 int replicated_rank = in RelayoutOperandsAndOutputs() local244 int replicated_rank = in RelayoutOperandsAndOutputs() local323 int replicated_rank = batchable_outputs_[output_index]; in ComputeLayoutForward() local359 int replicated_rank = batchable_operands_[operand_index]; in ComputeLayoutBackward() local