Home
last modified time | relevance | path

Searched refs:ArrayShape (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tools/
Ddriver.cc121 struct ArrayShape { struct
128 std::vector<ArrayShape> elements;
131 std::string ArrayShapeToString(ArrayShape shape) { in ArrayShapeToString()
145 ArrayShape ArrayShapeFromString(const std::string& s) { in ArrayShapeFromString()
172 std::vector<ArrayShape> out; in TupleShapeFromString()
325 int GetNumElements(const ArrayShape& shape) { in GetNumElements()
354 void Fill(void* buffer, const ArrayShape& shape) { in Fill()
402 void Display(const void* buffer, const ArrayShape& shape) { in Display()
439 ArrayShape array_shape = shape.elements[tuple_idx]; in Display()
474 ArrayShape shape = tuple_shape.elements[0]; in main()
/external/llvm-project/polly/include/polly/
DScopDetection.h70 struct ArrayShape { struct
77 ArrayShape(const SCEVUnknown *B) : BasePointer(B) {} in ArrayShape() argument
84 std::shared_ptr<ArrayShape> Shape;
89 MemAcc(const Instruction *I, std::shared_ptr<ArrayShape> S) in MemAcc()
266 std::shared_ptr<ArrayShape> Shape) const;
/external/llvm-project/polly/lib/Analysis/
DScopDetection.cpp973 std::shared_ptr<ArrayShape> Shape) const { in computeAccessFunctions()
1026 auto Shape = std::shared_ptr<ArrayShape>(new ArrayShape(BasePointer)); in hasBaseAffineAccesses()
/external/tensorflow/tensorflow/compiler/xla/
Dshape_tree_test.cc125 TEST_F(ShapeTreeTest, ArrayShape) { in TEST_F() argument