Searched refs:unknown_dims (Results 1 – 1 of 1) sorted by relevance
152 std::unordered_map<int64, int64>* unknown_dims) { in CompareSymbolicallyShapedTensorSizes() argument159 ++(*unknown_dims)[unknown_dim_id++]; in CompareSymbolicallyShapedTensorSizes()161 ++(*unknown_dims)[dim_size]; in CompareSymbolicallyShapedTensorSizes()170 std::set<int64> unknown_dims; in CompareSymbolicallyShapedTensorSizes() local171 for (const auto& el : left_unknown_dims) unknown_dims.insert(el.first); in CompareSymbolicallyShapedTensorSizes()172 for (const auto& el : right_unknown_dims) unknown_dims.insert(el.first); in CompareSymbolicallyShapedTensorSizes()175 for (int64 unknown_dim : unknown_dims) { in CompareSymbolicallyShapedTensorSizes()