Searched defs:reduction_tiling (Results 1 – 3 of 3) sorted by relevance
80 std::array<int64_t, 3> reduction_tiling = in RewriteReduction() local
665 const std::array<int64_t, 3>& reduction_tiling) { in ReductionIsRaceFree()
4534 std::array<int64_t, 3> reduction_tiling, const Shape& input_shape) { in CanVectorizeReduction()4578 Vector3 reduction_tiling = GetReductionTiling( in ComputeReductionCodegenInfo() local