Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_parser.cc297 bool ParseDimensionSizes(std::vector<int64>* dimension_sizes,
3394 bool HloParser::ParseDimensionSizes(std::vector<int64>* dimension_sizes, in ParseDimensionSizes() argument
3406 dimension_sizes->push_back(i); in ParseDimensionSizes()
3551 std::vector<int64> dimension_sizes; in ParseShape() local
3553 if (!ParseDimensionSizes(&dimension_sizes, &dynamic_dimensions)) { in ParseShape()
3557 for (int i = 0; i < dimension_sizes.size(); ++i) { in ParseShape()
3558 result->add_dimensions(dimension_sizes[i]); in ParseShape()