Home
last modified time | relevance | path

Searched refs:TupleShape (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tools/
Ddriver.cc127 struct TupleShape { struct
164 TupleShape TupleShapeFromString(std::string s) { in TupleShapeFromString()
182 std::string TupleShapeToString(TupleShape shape) { in TupleShapeToString()
204 std::map<int, TupleShape> buffers_shape;
302 TupleShape shape = TupleShapeFromString(output_shape); in ParseBufferAssignment()
314 TupleShape shape = TupleShapeFromString(param_shape); in ParseBufferAssignment()
432 void Display(const void* buffer, const TupleShape& shape) { in Display()
470 TupleShape tuple_shape = assignment.buffers_shape[allocation_idx]; in main()
/external/tensorflow/tensorflow/compiler/xla/
Dshape_tree_test.cc148 TEST_F(ShapeTreeTest, TupleShape) { in TEST_F() argument
/external/tensorflow/tensorflow/compiler/xla/service/
Dpattern_matcher_test.cc105 TEST_F(PatternMatcherTest, TupleShape) { in TEST_F() argument