Searched refs:ComputeArrayForReshape (Results 1 – 2 of 2) sorted by relevance
162 ComputeArrayForReshape(instr->shape(), in ComputeArrayFor()491 ComputeArrayForReshape(new_source_shape, operand->source())); in ReshapeToRemoveDegenerateDims()494 ComputeArrayForReshape(new_indices_shape, operand->indices())); in ReshapeToRemoveDegenerateDims()593 ComputeArrayForReshape(new_source_shape, operand->source())); in ReshapeToAddDegenerateDims()802 StatusOr<Analysis::Array*> IndexedArrayAnalysis::ComputeArrayForReshape( in ComputeArrayForReshape() function in xla::IndexedArrayAnalysis
322 StatusOr<Array*> ComputeArrayForReshape(const Shape& shape, Array* operand);