Home
last modified time | relevance | path

Searched defs:reduction_tiling (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dtree_reduction_rewriter.cc80 std::array<int64_t, 3> reduction_tiling = in RewriteReduction() local
H A Dir_emission_utils.cc665 const std::array<int64_t, 3>& reduction_tiling) { in ReductionIsRaceFree()
H A Dir_emitter_unnested.cc4534 std::array<int64_t, 3> reduction_tiling, const Shape& input_shape) { in CanVectorizeReduction()
4578 Vector3 reduction_tiling = GetReductionTiling( in ComputeReductionCodegenInfo() local