Home
last modified time | relevance | path

Searched refs:ShapeLayout (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dshape_layout.cc25 Status ShapeLayout::CopyLayoutFromShape(const Shape& other_shape) { in CopyLayoutFromShape()
35 Status ShapeLayout::AssignLayoutToShape(Shape* to_shape) const { in AssignLayoutToShape()
45 void ShapeLayout::SetToDefaultLayout() { in SetToDefaultLayout()
49 bool ShapeLayout::MatchesLayoutInShape(const Shape& shape, in MatchesLayoutInShape()
82 const Layout& ShapeLayout::layout() const { in layout()
88 void ShapeLayout::Clear() { LayoutUtil::ClearLayout(&shape_); } in Clear()
90 bool ShapeLayout::LayoutIsSet() const { return LayoutUtil::HasLayout(shape_); } in LayoutIsSet()
92 void ShapeLayout::ResetLayout(const Layout& layout) { in ResetLayout()
99 void ShapeLayout::ResetLayout(const Layout& layout, in ResetLayout()
107 bool ShapeLayout::operator==(const ShapeLayout& other) const { in operator ==()
[all …]
Dshape_layout.h34 class ShapeLayout {
38 explicit ShapeLayout(const Shape& shape) : shape_(shape) {} in ShapeLayout() function
91 bool operator==(const ShapeLayout& other) const;
92 bool operator!=(const ShapeLayout& other) const;
/external/tensorflow/tensorflow/compiler/xla/service/
Dcomputation_layout.h36 explicit ComputationLayout(ShapeLayout result_layout) in ComputationLayout()
46 void add_parameter_layout(ShapeLayout shape_layout) { in add_parameter_layout()
51 const ShapeLayout& parameter_layout(int64_t param_no) const { in parameter_layout()
54 ShapeLayout* mutable_parameter_layout(int64_t param_no) { in mutable_parameter_layout()
62 const std::vector<ShapeLayout>& parameter_layouts() const { in parameter_layouts()
67 const ShapeLayout& result_layout() const { return result_layout_; } in result_layout()
68 ShapeLayout* mutable_result_layout() { return &result_layout_; } in mutable_result_layout()
97 std::vector<ShapeLayout> parameter_layouts_;
98 ShapeLayout result_layout_;
Dlayout_assignment_test.cc109 const ShapeLayout shape_layout(shape); in TEST_F()
139 const ShapeLayout col_major(col_major_shape); in TEST_F()
144 const ShapeLayout row_major(row_major_shape); in TEST_F()
191 const ShapeLayout shape_layout(shape); in TEST_F()
373 ShapeLayout(ashape_with_layout); in TEST_F()
374 *computation_layout.mutable_result_layout() = ShapeLayout(bshape_with_layout); in TEST_F()
414 ShapeLayout(ashape_with_layout); in TEST_F()
415 *computation_layout.mutable_result_layout() = ShapeLayout(bshape_with_layout); in TEST_F()
452 ShapeLayout(input_shape_with_layout); in TEST_F()
454 ShapeLayout(output_shape_with_layout); in TEST_F()
[all …]
Dlayout_assignment.h107 const ShapeLayout& shape_layout, const HloInstruction* instruction,
111 const ShapeLayout& shape_layout() const { return shape_layout_; } in shape_layout()
121 ShapeLayout shape_layout_;
130 const ShapeLayout& shape_layout, bool dfs = false,
135 const ShapeLayout& shape_layout() const { return shape_layout_; } in shape_layout()
139 const ShapeLayout shape_layout_;
173 const ShapeLayout* OperandLayout(const HloInstruction* instruction,
177 const ShapeLayout* ResultLayout() const;
373 Status PropagateUseConstraintToDefs(const ShapeLayout& shape_layout,
498 virtual Status CopyOperandIfLayoutsDiffer(const ShapeLayout& operand_layout,
Dhlo_runner_interface.cc125 ShapeLayout(shape_representation_fn(shape)); in UpdateEntryComputationLayout()
128 ShapeLayout(shape_representation_fn( in UpdateEntryComputationLayout()
Dlayout_assignment.cc86 const ShapeLayout& shape_layout, const HloInstruction* instruction, in OperandLayoutConstraint()
275 OperandLayoutConstraint(ShapeLayout(shape_with_layout), instruction, in SetOperandLayout()
280 OperandLayoutConstraint(ShapeLayout(shape_with_layout), instruction, in SetOperandLayout()
315 const ShapeLayout* curr_shape_layout = ResultLayout(); in SetResultLayout()
329 new ResultLayoutConstraint(ShapeLayout(shape_with_layout), dfs)); in SetResultLayout()
383 const ShapeLayout* LayoutConstraints::OperandLayout( in OperandLayout()
398 const ShapeLayout* LayoutConstraints::ResultLayout() const { in ResultLayout()
539 const ShapeLayout& parameter_layout = in AddMandatoryConstraints()
744 const ShapeLayout& result_layout = computation_layout->result_layout(); in AddMandatoryConstraints()
852 const ShapeLayout& parameter_layout = in CheckParameterLayout()
[all …]
Dhlo_module_config.cc55 for (const ShapeLayout& param_layout : in compilation_cache_key()
Dcomputation_layout.cc60 [](const ShapeLayout& s) { return s.LayoutIsSet(); }) && in LayoutIsSet()
Dhlo_computation.cc149 ShapeLayout(instruction->shape())); in AddEntryComputationParameter()
169 param_no) = ShapeLayout(instruction->shape()); in ReplaceEntryComputationParameter()
Dhlo_verifier.cc1617 const ShapeLayout& result_layout = layout.result_layout(); in VerifyEntryComputationLayout()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_layout_assignment_test.cc80 ShapeLayout(LayoutUtil::GetWithDefaultLayout(lhs_shape)); in TEST_F()
82 ShapeLayout(LayoutUtil::GetWithDefaultLayout(result_shape)); in TEST_F()
121 ShapeLayout(LayoutUtil::GetWithDefaultLayout(lhs_shape)); in TEST_F()
123 ShapeLayout(LayoutUtil::GetWithDefaultLayout(result_shape)); in TEST_F()
165 ShapeLayout(LayoutUtil::GetWithDefaultLayout(lhs_a_shape)); in TEST_F()
167 ShapeLayout(LayoutUtil::GetWithDefaultLayout(lhs_b_shape)); in TEST_F()
169 ShapeLayout(LayoutUtil::GetWithDefaultLayout(tuple_result->shape())); in TEST_F()
199 ShapeLayout(LayoutUtil::GetWithDefaultLayout(rhs_shape)); in TEST_F()
201 ShapeLayout(LayoutUtil::GetWithDefaultLayout(result_shape)); in TEST_F()
239 ShapeLayout(LayoutUtil::GetWithDefaultLayout(lhs_shape)); in TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_layout_assignment_test.cc73 ShapeLayout(lhs_shape_with_layout); in TEST_F()
75 ShapeLayout(rhs_shape_with_layout); in TEST_F()
77 ShapeLayout(result_shape_with_layout); in TEST_F()
160 ShapeLayout(input_shape); in TEST_F()
164 *computation_layout.mutable_result_layout() = ShapeLayout(result_shape); in TEST_F()
228 ShapeLayout(input_shape); in TEST_F()
233 ShapeLayout(ShapeUtil::MakeTupleShape( in TEST_F()
309 ShapeLayout(input_shape); in TEST_F()
314 ShapeLayout(ShapeUtil::MakeTupleShape( in TEST_F()
/external/tensorflow/tensorflow/core/tpu/
Dtpu_execute.cc453 xla::ShapeLayout(xla::Shape(executable.output_shape()))); in TPUExecute()
456 computation_layout.add_parameter_layout(xla::ShapeLayout(shape)); in TPUExecute()
/external/tensorflow/tensorflow/stream_executor/tpu/
Dc_api_conversions.cc403 xla::ShapeLayout result_layout( in ConfigWithLayout()
407 layout.add_parameter_layout(xla::ShapeLayout( in ConfigWithLayout()
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_execute_op.cc204 const std::vector<xla::ShapeLayout>& compile_time_shapes) { in UpdateDynamicInputs()
350 const std::vector<xla::ShapeLayout>& shape_layouts = in RunExecutable()