Searched refs:unknown_dims (Results 1 – 1 of 1) sorted by relevance
155 std::unordered_map<int64, int64>* unknown_dims) { in CompareSymbolicallyShapedTensorSizes() argument162 ++(*unknown_dims)[unknown_dim_id++]; in CompareSymbolicallyShapedTensorSizes()164 ++(*unknown_dims)[dim_size]; in CompareSymbolicallyShapedTensorSizes()173 std::set<int64> unknown_dims; in CompareSymbolicallyShapedTensorSizes() local174 for (const auto& el : left_unknown_dims) unknown_dims.insert(el.first); in CompareSymbolicallyShapedTensorSizes()175 for (const auto& el : right_unknown_dims) unknown_dims.insert(el.first); in CompareSymbolicallyShapedTensorSizes()178 for (int64 unknown_dim : unknown_dims) { in CompareSymbolicallyShapedTensorSizes()