Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_instructions.cc2402 operand_shapes_with_layout_(operand_shapes_with_layout.begin(), in HloCustomCallInstruction()
2427 for (const Shape& shape : operand_shapes_with_layout_) { in ToProto()
2480 for (const Shape& shape : operand_shapes_with_layout_) { in ExtraAttributesToStringImpl()
2539 for (int64 i = 0; i < operand_shapes_with_layout_.size(); ++i) { in IdenticalSlowPath()
2540 if (!ShapeUtil::Equal(operand_shapes_with_layout_[i], in IdenticalSlowPath()
2541 casted_other.operand_shapes_with_layout_[i])) { in IdenticalSlowPath()
2589 cloned->operand_shapes_with_layout_ = operand_shapes_with_layout(); in CloneWithNewOperandsImpl()
Dhlo_instructions.h1489 return operand_shapes_with_layout_; in operand_shapes_with_layout()
1535 std::vector<Shape> operand_shapes_with_layout_; variable