Searched refs:dims_set (Results 1 – 3 of 3) sorted by relevance
616 std::set<int64_t> dims_set(dimensions_to_reduce.begin(), in __anonb720852f0502() local623 dims_set.size() != dimensions_to_reduce.size() || in __anonb720852f0502()666 std::set<int64_t> dims_set(dimensions_to_reduce.begin(), in __anonb720852f0702() local673 dims_set.size() != dimensions_to_reduce.size() || in __anonb720852f0702()746 std::set<int64_t> dims_set(dimensions_to_reduce.begin(), in __anonb720852f0902() local754 dims_set.size() != dimensions_to_reduce.size() || in __anonb720852f0902()762 if (dims_set.find(i) == dims_set.end()) { in __anonb720852f0902()
72 std::set<int64_t> dims_set(broadcast_dims.begin(), broadcast_dims.end()); in Compile() local74 dims_set.size() == broadcast_dims.size() && in Compile()
235 std::set<std::string> dims_set; in GetAbstractMesh() local237 if (dims_set.find(dim.name) != dims_set.end()) in GetAbstractMesh()244 dims_set.insert(dim.name); in GetAbstractMesh()692 std::set<std::string> dims_set; in GetLayout() local707 if (dims_set.find(sharding_spec) != dims_set.end()) in GetLayout()712 dims_set.insert(sharding_spec); in GetLayout()