Home
last modified time | relevance | path

Searched refs:IndexedShape (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dmemory_space_propagation.cc32 for (const ShapeUtil::IndexedShape& indexed_shape : in Run()
43 for (const ShapeUtil::IndexedShape& indexed_shape : in Run()
Dhlo_cost_analysis.cc1105 for (const ShapeUtil::IndexedShape& indexed_shape : in GetBytesRead()
1123 for (const ShapeUtil::IndexedShape& indexed_shape : in GetBytesWritten()
/external/tensorflow/tensorflow/compiler/xla/
Dshape_util.h187 struct IndexedShape { struct
188 IndexedShape() = default;
189 IndexedShape(ShapeIndex index, Shape shape) in IndexedShape() function
568 static std::vector<IndexedShape> GetLeafShapes(const Shape& shape);
Dshape_util.cc960 /* static */ std::vector<ShapeUtil::IndexedShape> ShapeUtil::GetLeafShapes( in GetLeafShapes()
962 std::vector<IndexedShape> leaves; in GetLeafShapes()
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client.cc204 for (const ShapeUtil::IndexedShape& indexed_shape : in FromLiterals()