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.cc2068 operand_shapes_with_layout_(operand_shapes_with_layout.begin(), in HloCustomCallInstruction()
2090 for (const Shape& shape : operand_shapes_with_layout_) { in ToProto()
2126 for (const Shape& shape : operand_shapes_with_layout_) { in ExtraAttributesToStringImpl()
2164 for (int64 i = 0; i < operand_shapes_with_layout_.size(); ++i) { in IdenticalSlowPath()
2165 if (!ShapeUtil::Equal(operand_shapes_with_layout_[i], in IdenticalSlowPath()
2166 casted_other.operand_shapes_with_layout_[i])) { in IdenticalSlowPath()
2183 cloned->operand_shapes_with_layout_ = operand_shapes_with_layout(); in CloneWithNewOperandsImpl()
Dhlo_instructions.h1221 return operand_shapes_with_layout_; in operand_shapes_with_layout()
1250 std::vector<Shape> operand_shapes_with_layout_; variable