Home
last modified time | relevance | path

Searched refs:operand_shapes_with_layout_ (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_instructions.cc2704 operand_shapes_with_layout_(operand_shapes_with_layout.begin(), in HloCustomCallInstruction()
2728 for (const Shape& shape : operand_shapes_with_layout_) { in ToProto()
2784 shape_strings.reserve(operand_shapes_with_layout_.size()); in ExtraAttributesToStringImpl()
2785 for (const Shape& shape : operand_shapes_with_layout_) { in ExtraAttributesToStringImpl()
2853 for (int64_t i = 0; i < operand_shapes_with_layout_.size(); ++i) { in IdenticalSlowPath()
2854 if (!ShapeUtil::Equal(operand_shapes_with_layout_[i], in IdenticalSlowPath()
2855 casted_other.operand_shapes_with_layout_[i])) { in IdenticalSlowPath()
2911 cloned->operand_shapes_with_layout_ = operand_shapes_with_layout(); in CloneWithNewOperandsImpl()
H A Dhlo_instructions.h1892 return operand_shapes_with_layout_; in operand_shapes_with_layout()
1957 std::vector<Shape> operand_shapes_with_layout_; variable