Searched refs:reduced_dimensions (Results 1 – 4 of 4) sorted by relevance
84 std::vector<int64> reduced_dimensions = hlo->dimensions(); in RewriteReduction() local85 absl::c_sort(reduced_dimensions); in RewriteReduction()86 CHECK_LE(reduced_dimensions.size(), 2); in RewriteReduction()88 reduced_dimensions[reduced_dimensions.size() - 1]; in RewriteReduction()
54 const std::vector<int64> &reduced_dimensions = instr->dimensions(); in HandleReduce() local62 if (absl::c_linear_search(reduced_dimensions, dim)) { in HandleReduce()
52 auto reduced_dimensions = hlo->dimensions(); in HandleReduce() local
2721 absl::Span<const int64> reduced_dimensions(reduce->dimensions()); in EmitElementalReduce() local2752 loops.AddLoopsForShapeOnDimensions(arg->shape(), reduced_dimensions, in EmitElementalReduce()