Searched defs:reduced_dims (Results 1 – 2 of 2) sorted by relevance
2992 absl::Span<const int64> reduced_dims = in EmitBlock() local3107 absl::Span<const int64> reduced_dims = in EmitKernel() local
1367 absl::flat_hash_set<int64> reduced_dims(reduce.dimensions().begin(), in ReductionPreservesLayout() local