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.cc40 for (const ShapeUtil::IndexedShape& indexed_shape : in Run()
51 for (const ShapeUtil::IndexedShape& indexed_shape : in Run()
Dhlo_cost_analysis.cc1191 for (const ShapeUtil::IndexedShape& indexed_shape : in GetBytesWritten()
/external/tensorflow/tensorflow/compiler/xla/
Dshape_util.h97 struct IndexedShape { struct
98 IndexedShape() = default;
99 IndexedShape(ShapeIndex index, Shape shape) in IndexedShape() argument
467 static std::vector<IndexedShape> GetLeafShapes(const Shape& shape);
Dshape_util.cc1039 /* static */ std::vector<ShapeUtil::IndexedShape> ShapeUtil::GetLeafShapes( in GetLeafShapes()
1041 std::vector<IndexedShape> leaves; in GetLeafShapes()
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client.cc215 for (const ShapeUtil::IndexedShape& indexed_shape : in FromLiterals()