Home
last modified time | relevance | path

Searched refs:ComputeReductionCodegenInfo (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dir_emitter_unnested.h446 StatusOr<ReductionCodegenInfo> ComputeReductionCodegenInfo(
H A Dir_emitter_unnested.cc4569 StatusOr<ReductionCodegenInfo> IrEmitterUnnested::ComputeReductionCodegenInfo( in ComputeReductionCodegenInfo() function in xla::gpu::IrEmitterUnnested
4926 ComputeReductionCodegenInfo(fusion, first_reduce)); in EmitUnnestedReduction()
4976 ComputeReductionCodegenInfo(fusion, first_reduce)); in EmitUnnestedReduction()