Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/
Dshape_util.h103 class ShapeIndexView {
105 ShapeIndexView(const ShapeIndex& shape_index, int64 offset = 0)
106 : ShapeIndexView(shape_index.data() + offset,
110 ShapeIndexView(std::initializer_list<int64> indices) in ShapeIndexView() function
111 : ShapeIndexView(indices.begin(), indices.end()) {} in ShapeIndexView()
112 ShapeIndexView(const ShapeIndexView& other) = default;
124 ShapeIndexView ConsumeFront() const { in ConsumeFront()
128 return ShapeIndexView(new_begin, end_); in ConsumeFront()
134 ShapeIndexView(iterator begin, iterator end) : begin_(begin), end_(end) {} in ShapeIndexView() function
141 std::ostream& operator<<(std::ostream& out, const ShapeIndexView& shape_index);
[all …]
Dshape_util.cc49 string ShapeIndexView::ToString() const { in ToString()
62 std::ostream& operator<<(std::ostream& out, const ShapeIndexView& shape_index) { in operator <<()
817 ShapeIndexView index) { in GetSubshape()
828 ShapeIndexView index) { in GetMutableSubshape()