Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_parser.cc2187 std::vector<int64> elems_seen_per_dim(rank); in ParseDenseLiteral() local
2188 auto get_index_str = [&elems_seen_per_dim](int dim) -> string { in ParseDenseLiteral()
2189 std::vector<int64> elems_seen_until_dim(elems_seen_per_dim.begin(), in ParseDenseLiteral()
2190 elems_seen_per_dim.begin() + dim); in ParseDenseLiteral()
2205 elems_seen_per_dim[rank - 1]++; in ParseDenseLiteral()
2206 if (elems_seen_per_dim[rank - 1] > shape.dimensions(rank - 1)) { in ParseDenseLiteral()
2227 elems_seen_per_dim[nest_level - 2]++; in ParseDenseLiteral()
2228 if (elems_seen_per_dim[nest_level - 2] > in ParseDenseLiteral()
2241 if (elems_seen_per_dim[nest_level] != shape.dimensions(nest_level)) { in ParseDenseLiteral()
2245 elems_seen_per_dim[nest_level])); in ParseDenseLiteral()
[all …]