Searched refs:slice_operand (Results 1 – 3 of 3) sorted by relevance
767 auto slice_operand = !windowed_op_is_lhs ? l : r; in PartitionBaseCase() local771 slice_operand->shape().dimensions(slice_sharding_dim); in PartitionBaseCase()773 slice_operand = PadBaseShapeBeforeUnevenTiledSharding( in PartitionBaseCase()774 slice_operand, *slice_sharding, &body_b); in PartitionBaseCase()782 for (int64 i = 0; i < slice_operand->shape().dimensions_size(); ++i) { in PartitionBaseCase()786 new_dims.push_back(slice_operand->shape().dimensions(i)); in PartitionBaseCase()790 ShapeUtil::MakeShape(slice_operand->shape().element_type(), in PartitionBaseCase()792 slice_operand)); in PartitionBaseCase()832 slice_operand->shape(), reshaped_slice_operand->shape(), in PartitionBaseCase()1075 auto slice_operand = !windowed_op_is_lhs ? l : r; in PartitionBaseCase() local[all …]
1630 std::vector<HloInstruction*> slice_operand; in ConvertToDynamic() local1631 slice_operand.push_back(inst); in ConvertToDynamic()1641 slice_operand.push_back(dimension_size); in ConvertToDynamic()1644 output_shape, slice_operand, "SliceToDynamic")); in ConvertToDynamic()
1253 HloInstruction* slice_operand = operands[i]->mutable_operand(0); in HandleConcatenate() local1257 operands[j]->operand(0) == slice_operand && in HandleConcatenate()1289 new_slice_shape, slice_operand, in HandleConcatenate()