| /external/tensorflow/tensorflow/compiler/xla/service/ |
| layout_assignment.h | 35 #include "tensorflow/compiler/xla/shape_layout.h"
92 OperandLayoutConstraint(const ShapeLayout& shape_layout,
96 const ShapeLayout& shape_layout() const { return shape_layout_; } function in class:xla::OperandLayoutConstraint
114 explicit ResultLayoutConstraint(const ShapeLayout& shape_layout,
117 shape_layout_(shape_layout) {}
119 const ShapeLayout& shape_layout() const { return shape_layout_; } function in class:xla::ResultLayoutConstraint
335 Status PropagateUseConstraintToDefs(const ShapeLayout& shape_layout,
|
| hlo_module_config.cc | 22 #include "tensorflow/compiler/xla/shape_layout.h"
|
| computation_layout.h | 23 #include "tensorflow/compiler/xla/shape_layout.h"
|
| layout_assignment.cc | 38 #include "tensorflow/compiler/xla/shape_layout.h"
153 const ShapeLayout& shape_layout, const HloInstruction* instruction,
156 shape_layout_(shape_layout),
160 CHECK(ShapeUtil::Compatible(shape_layout.shape(),
162 << shape_layout.shape() << " is not compatible with "
278 if (curr_shape_layout->shape_layout().MatchesLayoutInShape(
288 curr_shape_layout->shape_layout().ToString().c_str(),
409 return &constraint->shape_layout();
421 return result_constraint_ ? &result_constraint_->shape_layout() : nullptr;
[all...] |
| layout_assignment_test.cc | 32 #include "tensorflow/compiler/xla/shape_layout.h"
80 const ShapeLayout shape_layout(shape);
83 *computation_layout.mutable_parameter_layout(0) = shape_layout;
84 *computation_layout.mutable_parameter_layout(1) = shape_layout;
85 *computation_layout.mutable_result_layout() = shape_layout;
162 const ShapeLayout shape_layout(shape);
165 *computation_layout.mutable_result_layout() = shape_layout;
[all...] |
| local_service.cc | 36 #include "tensorflow/compiler/xla/shape_layout.h"
|
| BUILD | 546 "//tensorflow/compiler/xla:shape_layout",
581 "//tensorflow/compiler/xla:shape_layout",
[all...] |
| service.cc | 42 #include "tensorflow/compiler/xla/shape_layout.h"
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| gpu_layout_assignment.cc | 166 layout_constraint.shape_layout().shape(), instruction));
181 layout_constraint.shape_layout().layout(), *out_buf));
196 layout_constraint.shape_layout().layout(), *out_buf));
200 layout_constraint.shape_layout().shape(), instruction, operand_to_set));
|
| gpu_layout_assignment_test.cc | 25 #include "tensorflow/compiler/xla/shape_layout.h"
|
| BUILD | 594 "//tensorflow/compiler/xla:shape_layout",
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| BUILD | 568 name = "shape_layout",
569 srcs = ["shape_layout.cc"],
570 hdrs = ["shape_layout.h"],
|
| shape_layout.cc | 16 #include "tensorflow/compiler/xla/shape_layout.h"
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| hlo_test_base.h | 29 #include "tensorflow/compiler/xla/shape_layout.h"
|
| BUILD | 109 "//tensorflow/compiler/xla:shape_layout",
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| cpu_layout_assignment_test.cc | 32 #include "tensorflow/compiler/xla/shape_layout.h"
|
| BUILD | 678 "//tensorflow/compiler/xla:shape_layout",
|
