Searched refs:ShapeIndexView (Results 1 – 16 of 16) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.h | 121 class ShapeIndexView {
123 ShapeIndexView(const ShapeIndex& shape_index, int64 offset = 0)
127 ShapeIndexView(std::initializer_list<int64> indices) : indices_(indices) {} in ShapeIndexView() function
128 ShapeIndexView(const ShapeIndexView& other) = default;
140 ShapeIndexView ConsumeFront() const { in ConsumeFront()
141 ShapeIndexView result = *this; in ConsumeFront()
145 ShapeIndexView ConsumeBack() const { in ConsumeBack()
146 ShapeIndexView result = *this; in ConsumeBack()
152 bool operator==(const ShapeIndexView& other) const;
153 bool operator!=(const ShapeIndexView& other) const;
[all …]
|
| D | shape_tree.h | 119 const T& element(ShapeIndexView index) const;
120 T* mutable_element(ShapeIndexView index);
139 bool IsLeaf(ShapeIndexView index) const { return Lookup(index)->is_leaf; } in IsLeaf()
225 iterator find(ShapeIndexView index) { in find()
230 const_iterator find(ShapeIndexView index) const { in find()
318 Node* Lookup(ShapeIndexView index);
319 const Node* Lookup(ShapeIndexView index) const;
555 const T& ShapeTree<T>::element(ShapeIndexView index) const { in element()
560 T* ShapeTree<T>::mutable_element(ShapeIndexView index) { in mutable_element()
565 internal::ShapeTreeNode<T>* ShapeTree<T>::Lookup(ShapeIndexView index) { in Lookup()
[all …]
|
| D | shape_util.cc | 54 string ShapeIndex::ToString() const { return ShapeIndexView(*this).ToString(); } in ToString()
56 string ShapeIndexView::ToString() const { in ToString()
60 bool ShapeIndexView::operator==(const ShapeIndexView& other) const { in operator ==()
64 bool ShapeIndexView::operator!=(const ShapeIndexView& other) const { in operator !=()
73 std::ostream& operator<<(std::ostream& out, const ShapeIndexView& shape_index) { in operator <<()
78 bool ShapeIndexView::StartsWith(ShapeIndexView prefix) const { in StartsWith()
770 ShapeIndexView index) { in IndexIsValid()
782 ShapeIndexView index) { in GetSubshape()
793 const Shape& shape, ShapeIndexView index) { in TryGetSubshape()
808 ShapeIndexView index) { in GetMutableSubshape()
|
| D | shape_layout.h | 77 void ResetLayout(const Layout& layout, ShapeIndexView shape_index);
|
| D | shape_layout.cc | 71 ShapeIndexView shape_index) { in ResetLayout()
|
| D | shape_util_test.cc | 36 ShapeIndexView index_view(index, 1); in TEST()
38 EXPECT_EQ(ShapeIndexView({2, 3, 4}), index_view); in TEST()
39 EXPECT_EQ(ShapeIndexView({3, 4}), index_view.ConsumeFront()); in TEST()
40 EXPECT_EQ(ShapeIndexView({2, 3}), index_view.ConsumeBack()); in TEST()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | hlo_to_ir_bindings.h | 54 ShapeIndexView shape_index = {});
74 ShapeIndexView shape_index = {}) const {
100 ShapeIndexView shape_index,
|
| D | infeed_thunk.cc | 61 ShapeIndexView(index, 1)); in ExecuteOnStream()
70 infeed_buffers.mutable_element(ShapeIndexView(index, 1)); in ExecuteOnStream()
|
| D | hlo_to_ir_bindings.cc | 154 ShapeIndexView shape_index, in GetTypedIrValue()
178 ShapeIndexView shape_index) { in BindHloToIrValue()
|
| D | outfeed_thunk.cc | 65 outfeed_slices_.element(ShapeIndexView(index).ConsumeFront()); in ExecuteOnStream()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shaped_buffer.h | 182 ScopedShapedBuffer TakeSubTree(ShapeIndexView index);
|
| D | shaped_buffer.cc | 160 ScopedShapedBuffer ScopedShapedBuffer::TakeSubTree(ShapeIndexView index) { in TakeSubTree()
|
| D | shaped_buffer_test.cc | 143 if (ShapeIndexView(index).StartsWith(subtree_index)) { in TEST()
|
| D | pattern_matcher.h | 900 ShapeIndexView index,
939 ShapeIndexView index_;
1070 auto WithSubshape(ShapeIndexView index,
1085 WithSubshapeEqualTo(ShapeIndexView index, const ::xla::Shape* shape) const {
1098 WithSubshapeCompatibleTo(ShapeIndexView index,
|
| D | dynamic_dimension_inference.cc | 116 ShapeIndexView(index).ConsumeFront().ToShapeIndex(); in HandleGetTupleElement()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | hlo_test_base.h | 282 ShapeIndexView shape_index) { in ForceResultLayout()
|