HomeSort by relevance Sort by last modified time
    Searched refs:ShapeLayout (Results 1 - 10 of 10) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/
shape_layout.cc 25 tensorflow::Status ShapeLayout::CopyLayoutFromShape(const Shape& other_shape) {
35 tensorflow::Status ShapeLayout::AssignLayoutToShape(Shape* to_shape) const {
45 void ShapeLayout::SetToDefaultLayout() {
49 bool ShapeLayout::MatchesLayoutInShape(const Shape& shape) const {
53 const Layout& ShapeLayout::layout() const {
59 void ShapeLayout::Clear() { LayoutUtil::ClearLayout(&shape_); }
61 bool ShapeLayout::LayoutIsSet() const { return LayoutUtil::HasLayout(shape_); }
63 void ShapeLayout::ResetLayout(const Layout& layout) {
70 bool ShapeLayout::operator==(const ShapeLayout& other) const
    [all...]
shape_layout.h 28 // A ShapeLayout object encapsulates the layout of a particular shape (including
30 // single array. ShapeLayout contains a Layout proto for each array in the shape
32 // trivially holds a single Layout. Logically, ShapeLayout holds a nonmutable
34 class ShapeLayout {
36 // Constructs a ShapeLayout of the given shape. Layouts are copied from the
38 explicit ShapeLayout(const Shape& shape) : shape_(shape) {}
40 // Assigns the layouts in this ShapeLayout to the Layout fields of the given
41 // shape. 'to_shape' and the shape of the ShapeLayout object must be
45 // Returns true if the Layouts in this ShapeLayout match the layouts in the
47 // with the ShapeLayout's shape, then false is returned
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/
computation_layout.h 41 const ShapeLayout& parameter_layout(int64 param_no) const {
44 ShapeLayout* mutable_parameter_layout(int64 param_no) {
52 const std::vector<ShapeLayout>& parameter_layouts() const {
56 // Returns the ShapeLayout of a result of the computation.
57 const ShapeLayout& result_layout() const { return result_layout_; }
58 ShapeLayout* mutable_result_layout() { return &result_layout_; }
77 std::vector<ShapeLayout> parameter_layouts_;
78 ShapeLayout result_layout_;
layout_assignment.h 92 OperandLayoutConstraint(const ShapeLayout& shape_layout,
96 const ShapeLayout& shape_layout() const { return shape_layout_; }
106 ShapeLayout shape_layout_;
114 explicit ResultLayoutConstraint(const ShapeLayout& shape_layout,
119 const ShapeLayout& shape_layout() const { return shape_layout_; }
123 const ShapeLayout shape_layout_;
157 const ShapeLayout* OperandLayout(const HloInstruction* instruction,
161 const ShapeLayout* ResultLayout() const;
335 Status PropagateUseConstraintToDefs(const ShapeLayout& shape_layout,
computation_layout.cc 43 [](const ShapeLayout& s) { return s.LayoutIsSet(); }) &&
hlo_module_config.cc 46 for (const ShapeLayout& param_layout :
layout_assignment_test.cc 80 const ShapeLayout shape_layout(shape);
110 const ShapeLayout col_major(col_major_shape);
115 const ShapeLayout row_major(row_major_shape);
162 const ShapeLayout shape_layout(shape);
345 ShapeLayout(ashape_with_layout);
346 *computation_layout.mutable_result_layout() = ShapeLayout(bshape_with_layout);
386 ShapeLayout(ashape_with_layout);
387 *computation_layout.mutable_result_layout() = ShapeLayout(bshape_with_layout);
424 ShapeLayout(input_shape_with_layout);
426 ShapeLayout(output_shape_with_layout)
    [all...]
layout_assignment.cc 113 Status CopyOperandIfLayoutsDiffer(const ShapeLayout& operand_layout,
153 const ShapeLayout& shape_layout, const HloInstruction* instruction,
307 key, OperandLayoutConstraint(ShapeLayout(shape_with_layout),
312 OperandLayoutConstraint(ShapeLayout(shape_with_layout), instruction,
336 const ShapeLayout* curr_shape_layout = ResultLayout();
350 new ResultLayoutConstraint(ShapeLayout(shape_with_layout), dfs));
405 const ShapeLayout* LayoutConstraints::OperandLayout(
420 const ShapeLayout* LayoutConstraints::ResultLayout() const {
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
gpu_layout_assignment_test.cc 66 ShapeLayout(lhs_shape_with_layout);
68 ShapeLayout(rhs_shape_with_layout);
70 ShapeLayout(result_shape_with_layout);
152 ShapeLayout(input_shape);
156 *computation_layout.mutable_result_layout() = ShapeLayout(result_shape);
219 ShapeLayout(input_shape);
224 ShapeLayout(ShapeUtil::MakeTupleShape(
299 ShapeLayout(input_shape);
304 ShapeLayout(ShapeUtil::MakeTupleShape(
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_layout_assignment_test.cc 74 ShapeLayout(LayoutUtil::GetWithDefaultLayout(lhs_shape));
76 ShapeLayout(LayoutUtil::GetWithDefaultLayout(result_shape));
115 ShapeLayout(LayoutUtil::GetWithDefaultLayout(lhs_shape));
117 ShapeLayout(LayoutUtil::GetWithDefaultLayout(result_shape));
159 ShapeLayout(LayoutUtil::GetWithDefaultLayout(lhs_a_shape));
161 ShapeLayout(LayoutUtil::GetWithDefaultLayout(lhs_b_shape));
163 ShapeLayout(LayoutUtil::GetWithDefaultLayout(tuple_result->shape()));
193 ShapeLayout(LayoutUtil::GetWithDefaultLayout(rhs_shape));
195 ShapeLayout(LayoutUtil::GetWithDefaultLayout(result_shape));
233 ShapeLayout(LayoutUtil::GetWithDefaultLayout(lhs_shape))
    [all...]

Completed in 636 milliseconds