Searched refs:shape_layout (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | layout_assignment.h | 96 OperandLayoutConstraint(const ShapeLayout& shape_layout,
100 const ShapeLayout& shape_layout() const { return shape_layout_; } in shape_layout() function
118 explicit ResultLayoutConstraint(const ShapeLayout& shape_layout,
121 shape_layout_(shape_layout) {} in LayoutConstraint()
123 const ShapeLayout& shape_layout() const { return shape_layout_; } in shape_layout() function
359 Status PropagateUseConstraintToDefs(const ShapeLayout& shape_layout,
|
| D | computation_layout.h | 46 void add_parameter_layout(ShapeLayout shape_layout) { in add_parameter_layout() argument
47 parameter_layouts_.push_back(std::move(shape_layout)); in add_parameter_layout()
|
| D | layout_assignment.cc | 83 const ShapeLayout& shape_layout, const HloInstruction* instruction, in OperandLayoutConstraint() argument
86 shape_layout_(shape_layout), in OperandLayoutConstraint()
90 CHECK(ShapeUtil::Compatible(shape_layout.shape(), in OperandLayoutConstraint()
92 << shape_layout.shape() << " is not compatible with " in OperandLayoutConstraint()
219 if (curr_shape_layout->shape_layout().MatchesLayoutInShape( in SetOperandLayout()
229 curr_shape_layout->shape_layout().ToString(), in SetOperandLayout()
349 return &constraint->shape_layout(); in OperandLayout()
361 return result_constraint_ ? &result_constraint_->shape_layout() : nullptr; in ResultLayout()
1253 const ShapeLayout& shape_layout, const HloInstruction* instruction, in PropagateUseConstraintToDefs() argument
1260 [&shape_layout, constraints]( in PropagateUseConstraintToDefs()
[all …]
|
| D | layout_assignment_test.cc | 109 const ShapeLayout shape_layout(shape); in TEST_F() local
112 *computation_layout.mutable_parameter_layout(0) = shape_layout; in TEST_F()
113 *computation_layout.mutable_parameter_layout(1) = shape_layout; in TEST_F()
114 *computation_layout.mutable_result_layout() = shape_layout; in TEST_F()
191 const ShapeLayout shape_layout(shape); in TEST_F() local
194 *computation_layout.mutable_result_layout() = shape_layout; in TEST_F()
|
| D | BUILD | 962 "//tensorflow/compiler/xla:shape_layout",
999 "//tensorflow/compiler/xla:shape_layout",
3542 "//tensorflow/compiler/xla:shape_layout",
3971 "//tensorflow/compiler/xla:shape_layout",
4340 "//tensorflow/compiler/xla:shape_layout",
4355 "//tensorflow/compiler/xla:shape_layout",
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_layout_assignment.cc | 319 layout_constraint.shape_layout().shape(), instruction)); in PropagateOperandConstraint()
334 layout_constraint.shape_layout().layout(), *out_buf)); in PropagateOperandConstraint()
349 layout_constraint.shape_layout().layout(), *out_buf)); in PropagateOperandConstraint()
353 layout_constraint.shape_layout().shape(), instruction, operand_to_set)); in PropagateOperandConstraint()
|
| D | BUILD | 1041 "//tensorflow/compiler/xla:shape_layout",
1531 "//tensorflow/compiler/xla:shape_layout",
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | BUILD | 803 name = "shape_layout",
804 srcs = ["shape_layout.cc"],
805 hdrs = ["shape_layout.h"],
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | BUILD | 223 "//tensorflow/compiler/xla:shape_layout",
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | BUILD | 965 "//tensorflow/compiler/xla:shape_layout",
1063 "//tensorflow/compiler/xla:shape_layout",
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | BUILD | 160 "//tensorflow/compiler/xla:shape_layout",
|