Searched refs:update_dims (Results 1 – 2 of 2) sorted by relevance
124 absl::Span<const int64> update_dims, in DynamicAddSlice() argument127 xla::XlaOp current = xla::DynamicSlice(operand, start_indices, update_dims); in DynamicAddSlice()
115 auto update_dims = in UpdateDDNums() local129 update_dims(new_ddnums->mutable_lhs_contracting_dimensions()); in UpdateDDNums()130 update_dims(new_ddnums->mutable_lhs_batch_dimensions()); in UpdateDDNums()132 update_dims(new_ddnums->mutable_rhs_contracting_dimensions()); in UpdateDDNums()133 update_dims(new_ddnums->mutable_rhs_batch_dimensions()); in UpdateDDNums()