Searched refs:passthrough_tile (Results 1 – 1 of 1) sorted by relevance
809 std::vector<int64> passthrough_tile(update_or_gather_shape.rank(), 1); in PassthroughOperandToGatherOutputOrScatterUpdate() local830 passthrough_tile[offset_dim] = dim_partitions; in PassthroughOperandToGatherOutputOrScatterUpdate()833 passthrough_tile.push_back( in PassthroughOperandToGatherOutputOrScatterUpdate()837 tile_assignment.Reshape(passthrough_tile); in PassthroughOperandToGatherOutputOrScatterUpdate()854 std::vector<int64> passthrough_tile(operand_shape.rank(), 1); in PassthroughGatherOutputOrScatterUpdateToOperand() local873 passthrough_tile[i] = dim_partitions; in PassthroughGatherOutputOrScatterUpdateToOperand()877 passthrough_tile.push_back( in PassthroughGatherOutputOrScatterUpdateToOperand()881 if (tile_assignment.num_elements() != Product(passthrough_tile)) { in PassthroughGatherOutputOrScatterUpdateToOperand()884 tile_assignment.Reshape(passthrough_tile); in PassthroughGatherOutputOrScatterUpdateToOperand()