Home
last modified time | relevance | path

Searched refs:tuple_shapes_ (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dshape.h49 tuple_shapes_(std::move(tuple_shapes)) {} in Shape()
144 int tuple_shapes_size() const { return tuple_shapes_.size(); } in tuple_shapes_size()
145 const Shape& tuple_shapes(int index) const { return tuple_shapes_.at(index); } in tuple_shapes()
146 Shape* mutable_tuple_shapes(int index) { return &tuple_shapes_.at(index); } in mutable_tuple_shapes()
148 tuple_shapes_.push_back(Shape()); in add_tuple_shapes()
149 return &tuple_shapes_.back(); in add_tuple_shapes()
151 void clear_tuple_shapes() { tuple_shapes_.clear(); } in clear_tuple_shapes()
152 const std::vector<Shape>& tuple_shapes() const { return tuple_shapes_; } in tuple_shapes()
153 std::vector<Shape>* mutable_tuple_shapes() { return &tuple_shapes_; } in mutable_tuple_shapes()
178 for (auto& subshape : tuple_shapes_) { in clear_dynamic_dimensions()
[all …]
H A Dshape.cc49 tuple_shapes_.reserve(shape_proto.tuple_shapes_size()); in Shape()
51 tuple_shapes_.emplace_back(element_shape); in Shape()
104 return absl::c_any_of(tuple_shapes_, in IsInteger()
113 for (const Shape& subshape : tuple_shapes_) { in is_static()