Searched refs:ShapeIndexView (Results 1 – 18 of 18) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.h | 47 class ShapeIndexView; variable
75 explicit ShapeIndex(ShapeIndexView v);
126 class ShapeIndexView {
128 ShapeIndexView(const ShapeIndex& shape_index, int64 offset = 0)
132 ShapeIndexView(std::initializer_list<int64> indices) : indices_(indices) {} in ShapeIndexView() function
133 ShapeIndexView(const ShapeIndexView& other) = default;
149 ShapeIndexView ConsumeFront() const { in ConsumeFront()
150 ShapeIndexView result = *this; in ConsumeFront()
154 ShapeIndexView ConsumeBack() const { in ConsumeBack()
155 ShapeIndexView result = *this; in ConsumeBack()
[all …]
|
| D | shape_tree.h | 121 const T& element(ShapeIndexView index) const;
122 T* mutable_element(ShapeIndexView index);
142 bool IsLeaf(ShapeIndexView index) const { return Lookup(index)->is_leaf; } in IsLeaf()
226 iterator find(ShapeIndexView index) { in find()
231 const_iterator find(ShapeIndexView index) const { in find()
321 Node* Lookup(ShapeIndexView index);
322 const Node* Lookup(ShapeIndexView index) const;
597 const T& ShapeTree<T>::element(ShapeIndexView index) const { in element()
602 T* ShapeTree<T>::mutable_element(ShapeIndexView index) { in mutable_element()
607 internal::ShapeTreeNode<T>* ShapeTree<T>::Lookup(ShapeIndexView index) { in Lookup()
[all …]
|
| D | shape_util.cc | 83 string ShapeIndex::ToString() const { return ShapeIndexView(*this).ToString(); } in ToString()
85 string ShapeIndexView::ToString() const { in ToString()
89 bool ShapeIndexView::operator==(const ShapeIndexView& other) const { in operator ==()
93 bool ShapeIndexView::operator!=(const ShapeIndexView& other) const { in operator !=()
102 std::ostream& operator<<(std::ostream& out, const ShapeIndexView& shape_index) { in operator <<()
107 bool ShapeIndexView::StartsWith(ShapeIndexView prefix) const { in StartsWith()
403 ShapeIndexView index, in UpdateDynamicDimension()
897 ShapeIndexView index) { in IndexIsValid()
909 ShapeIndexView index) { in GetSubshape()
920 const Shape& shape, ShapeIndexView index) { in TryGetSubshape()
[all …]
|
| D | shape_layout.h | 82 void ResetLayout(const Layout& layout, ShapeIndexView shape_index);
|
| D | shape_layout.cc | 76 ShapeIndexView shape_index) { in ResetLayout()
|
| D | shape_util_test.cc | 39 ShapeIndexView index_view(index, 1); in TEST()
41 EXPECT_EQ(ShapeIndexView({2, 3, 4}), index_view); in TEST()
42 EXPECT_EQ(ShapeIndexView({3, 4}), index_view.ConsumeFront()); in TEST()
43 EXPECT_EQ(ShapeIndexView({2, 3}), index_view.ConsumeBack()); in TEST()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | hlo_to_ir_bindings.h | 52 ShapeIndexView shape_index = {});
72 ShapeIndexView shape_index = {}) const {
98 ShapeIndexView shape_index,
|
| D | hlo_to_ir_bindings.cc | 141 ShapeIndexView shape_index, in GetTypedIrValue()
156 ShapeIndexView shape_index) { in BindHloToIrValue()
|
| D | ir_emitter.h | 132 ShapeIndexView shape_index = {}) const {
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | memory_space_propagation.h | 37 bool Propagate(ShapeIndexView index, const HloInstruction* callee_instruction,
|
| D | memory_space_propagation.cc | 56 bool MemorySpacePropagation::Propagate(ShapeIndexView index, in Propagate()
|
| D | shaped_buffer.h | 201 ScopedShapedBuffer TakeSubTree(ShapeIndexView index);
|
| D | shaped_buffer.cc | 175 ScopedShapedBuffer ScopedShapedBuffer::TakeSubTree(ShapeIndexView index) { in TakeSubTree()
|
| D | shaped_buffer_test.cc | 146 if (ShapeIndexView(index).StartsWith(subtree_index)) { in TEST()
|
| D | pattern_matcher.h | 881 ShapeIndexView index,
920 ShapeIndexView index_;
1033 ShapeIndexView index,
1045 WithSubshapeEqualTo(ShapeIndexView index, const ::xla::Shape* shape) const {
1058 WithSubshapeCompatibleTo(ShapeIndexView index,
|
| D | dynamic_dimension_inference.cc | 218 ShapeIndexView(index).ConsumeFront().ToShapeIndex(); in HandleGetTupleElement()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | hlo_test_base.h | 283 ShapeIndexView shape_index) { in ForceResultLayout()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 2598 ShapeUtil::GetSubshape(shard_shape, ShapeIndexView(index, 1)); in HandleInfeed()
3007 per_branch_partitioned_shapes[i], ShapeIndexView(index)); in HandleOutfeed()
|