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

  /external/tensorflow/tensorflow/compiler/xla/
shape_util.h 175 struct IndexedShape {
176 IndexedShape() = default;
177 IndexedShape(ShapeIndex index, Shape shape)
506 static std::vector<IndexedShape> GetLeafShapes(const Shape& shape);
    [all...]
shape_util.cc 833 /* static */ std::vector<ShapeUtil::IndexedShape> ShapeUtil::GetLeafShapes(
835 std::vector<IndexedShape> leaves;
    [all...]

Completed in 98 milliseconds