Searched defs:dims_set (Results 1 – 3 of 3) sorted by relevance
72 std::set<int64> dims_set(broadcast_dims.begin(), broadcast_dims.end()); in Compile() local
36 std::set<int64> dims_set(dimensions_to_reduce_.begin(), in XlaReduceOp() local
540 std::set<int64> dims_set(dimensions_to_reduce.begin(), in __anonbbc1c2d10502() local590 std::set<int64> dims_set(dimensions_to_reduce.begin(), in __anonbbc1c2d10702() local