Searched refs:SameShape (Results 1 – 2 of 2) sorted by relevance
94 TEST_F(SparseDenseCDivTest, SameShape) { in TEST_F() argument
347 bool SameShape(const HloInstruction* lhs, const HloInstruction* rhs) const;454 bool AlgebraicSimplifierVisitor::SameShape(const HloInstruction* lhs, in SameShape() function in xla::AlgebraicSimplifierVisitor481 if (!SameShape(old_instruction, new_instruction)) { in ReplaceInstructionIfSameShape()2607 if (SameShape(reshape, operand)) { in HandleReshape()2882 if (SameShape(operand, dynamic_slice)) { in HandleDynamicSlice()2894 if (SameShape(dynamic_update_slice, update)) { in HandleDynamicUpdateSlice()3681 if (!SameShape(min_operand, max_operand)) { in TransformToClampIfSameShape()