Searched refs:ShapeIndexView (Results 1 – 23 of 23) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_tree.h | 57 const Entry& operator[](ShapeIndexView index) const;
124 const T& element(ShapeIndexView index) const { return find(index)->second; } in element()
125 T* mutable_element(ShapeIndexView index) { return &find(index)->second; } in mutable_element()
145 bool IsLeaf(ShapeIndexView index) const { in IsLeaf()
203 iterator find(ShapeIndexView index) { in find()
206 const_iterator find(ShapeIndexView index) const { in find()
277 auto without_prefix = ShapeIndexView(index).subspan(src_index.size()); in CopySubtreeFrom()
301 auto without_prefix = ShapeIndexView(it->first).subspan(index.size()); in SubShapeTree()
|
| D | shape_util.h | 55 using ShapeIndexView = absl::Span<const int64_t>; variable
77 explicit ShapeIndex(ShapeIndexView view) in ShapeIndex()
296 static void UpdateDynamicDimension(Shape* shape, ShapeIndexView index,
448 static bool IndexIsValid(const Shape& shape, ShapeIndexView index);
453 static const Shape& GetSubshape(const Shape& shape, ShapeIndexView index);
455 ShapeIndexView index);
456 static Shape* GetMutableSubshape(Shape* shape, ShapeIndexView index);
|
| D | shape_layout.h | 85 void ResetLayout(const Layout& layout, ShapeIndexView shape_index);
|
| D | shape_tree.cc | 44 const IndexTable::Entry& IndexTable::operator[](ShapeIndexView index) const { in operator []()
|
| D | shape_layout.cc | 100 ShapeIndexView shape_index) { in ResetLayout()
|
| D | shape_util.cc | 446 ShapeIndexView index, in UpdateDynamicDimension()
976 ShapeIndexView index) { in IndexIsValid()
988 ShapeIndexView index) { in GetSubshape()
999 const Shape& shape, ShapeIndexView index) { in TryGetSubshape()
1014 ShapeIndexView index) { in GetMutableSubshape()
|
| D | literal.h | 1098 Literal SubLiteral(ShapeIndexView shape_index);
|
| D | literal.cc | 511 Literal Literal::SubLiteral(ShapeIndexView shape_index) { in SubLiteral()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | hlo_to_ir_bindings.h | 44 ShapeIndexView shape_index = {});
64 ShapeIndexView shape_index = {}) const {
90 ShapeIndexView shape_index,
|
| D | hlo_to_ir_bindings.cc | 142 ShapeIndexView shape_index, in GetTypedIrValue()
157 ShapeIndexView shape_index) { in BindHloToIrValue()
|
| D | ir_emitter.h | 121 ShapeIndexView shape_index = {}) const {
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | memory_space_propagation.h | 40 bool Propagate(ShapeIndexView index, const HloInstruction* callee_instruction,
|
| D | memory_space_propagation.cc | 64 bool MemorySpacePropagation::Propagate(ShapeIndexView index, in Propagate()
|
| D | dynamic_dimension_inference.h | 79 ShapeIndexView index = {}) const;
|
| D | shaped_buffer.h | 200 ScopedShapedBuffer TakeSubTree(ShapeIndexView index);
|
| D | shaped_buffer.cc | 175 ScopedShapedBuffer ScopedShapedBuffer::TakeSubTree(ShapeIndexView index) { in TakeSubTree()
|
| D | shaped_buffer_test.cc | 149 ShapeIndexView(index).first(subtree_index.size()) == subtree_index) { in TEST()
|
| D | hlo_value.h | 287 ShapeIndexView input_index);
|
| D | hlo_value.cc | 287 ShapeIndexView input_index) { in AssignUnionOf()
|
| D | pattern_matcher.h | 935 ShapeIndexView index,
974 ShapeIndexView index_;
1092 ShapeIndexView index,
1104 WithSubshapeEqualTo(ShapeIndexView index, const ::xla::Shape* shape) const {
1117 WithSubshapeCompatibleTo(ShapeIndexView index,
|
| D | dynamic_dimension_inference.cc | 249 ShapeIndex new_index(ShapeIndexView(index).subspan(1)); in HandleGetTupleElement()
2017 HloInstruction* inst, ShapeIndexView index) const { in HasDynamicDimension()
2024 if (ShapeIndexView(subindex).first(index.size()) != index) { in HasDynamicDimension()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | hlo_test_base.h | 318 ShapeIndexView shape_index) { in ForceResultLayout()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 3354 shard_shape, ShapeIndexView(index).subspan(1)); in HandleInfeed()
3789 per_branch_partitioned_shapes[i], ShapeIndexView(index)); in HandleOutfeed()
|