Searched refs:dims_set (Results 1 – 3 of 3) sorted by relevance
35 std::set<int64> dims_set(dimensions_to_reduce_.begin(), in XlaReduceOp() local38 context, dims_set.size() == dimensions_to_reduce_.size(), in XlaReduceOp()
72 std::set<int64> dims_set(broadcast_dims.begin(), broadcast_dims.end()); in Compile() local74 dims_set.size() == broadcast_dims.size() && in Compile()
439 std::set<int64> dims_set(dimensions_to_reduce.begin(), in __anon00ae58f50402() local445 dims_set.size() != dimensions_to_reduce.size() || in __anon00ae58f50402()