HomeSort by relevance Sort by last modified time
    Searched refs:ShapeIndexView (Results 1 - 16 of 16) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
hlo_to_ir_bindings.h 54 ShapeIndexView shape_index = {});
74 ShapeIndexView shape_index = {}) const {
100 ShapeIndexView shape_index,
infeed_thunk.cc 61 ShapeIndexView(index, 1));
70 infeed_buffers.mutable_element(ShapeIndexView(index, 1));
outfeed_thunk.cc 65 outfeed_slices_.element(ShapeIndexView(index).ConsumeFront());
hlo_to_ir_bindings.cc 154 ShapeIndexView shape_index,
178 ShapeIndexView shape_index) {
  /external/tensorflow/tensorflow/compiler/xla/
shape_layout.h 77 void ResetLayout(const Layout& layout, ShapeIndexView shape_index);
shape_util.h 117 // NB! ShapeIndexView does not own the memory backing the index array.
119 // that lives longer than the ShapeIndexView instances pointing into
121 class ShapeIndexView {
123 ShapeIndexView(const ShapeIndex& shape_index, int64 offset = 0)
127 ShapeIndexView(std::initializer_list<int64> indices) : indices_(indices) {}
128 ShapeIndexView(const ShapeIndexView& other) = default;
140 ShapeIndexView ConsumeFront() const {
141 ShapeIndexView result = *this;
145 ShapeIndexView ConsumeBack() const
    [all...]
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; }
225 iterator find(ShapeIndexView index) {
230 const_iterator find(ShapeIndexView index) const {
318 Node* Lookup(ShapeIndexView index);
319 const Node* Lookup(ShapeIndexView index) const;
555 const T& ShapeTree<T>::element(ShapeIndexView index) const {
560 T* ShapeTree<T>::mutable_element(ShapeIndexView index) {
565 internal::ShapeTreeNode<T>* ShapeTree<T>::Lookup(ShapeIndexView index)
    [all...]
shape_layout.cc 71 ShapeIndexView shape_index) {
shape_util.cc 54 string ShapeIndex::ToString() const { return ShapeIndexView(*this).ToString(); }
56 string ShapeIndexView::ToString() const {
60 bool ShapeIndexView::operator==(const ShapeIndexView& other) const {
64 bool ShapeIndexView::operator!=(const ShapeIndexView& other) const {
73 std::ostream& operator<<(std::ostream& out, const ShapeIndexView& shape_index) {
78 bool ShapeIndexView::StartsWith(ShapeIndexView prefix) const {
770 ShapeIndexView index)
    [all...]
shape_util_test.cc 36 ShapeIndexView index_view(index, 1);
38 EXPECT_EQ(ShapeIndexView({2, 3, 4}), index_view);
39 EXPECT_EQ(ShapeIndexView({3, 4}), index_view.ConsumeFront());
40 EXPECT_EQ(ShapeIndexView({2, 3}), index_view.ConsumeBack());
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/
shaped_buffer.h 182 ScopedShapedBuffer TakeSubTree(ShapeIndexView index);
shaped_buffer.cc 160 ScopedShapedBuffer ScopedShapedBuffer::TakeSubTree(ShapeIndexView index) {
shaped_buffer_test.cc 143 if (ShapeIndexView(index).StartsWith(subtree_index)) {
pattern_matcher.h 900 ShapeIndexView index,
939 ShapeIndexView index_;
    [all...]
dynamic_dimension_inference.cc 116 ShapeIndexView(index).ConsumeFront().ToShapeIndex();
  /external/tensorflow/tensorflow/compiler/xla/tests/
hlo_test_base.h 282 ShapeIndexView shape_index) {

Completed in 362 milliseconds