Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc5407 if (auto unnested_reduce = mlir::dyn_cast<mlir::lmhlo::ReduceOp>(reduce)) { in EmitIRForReduction() local
5409 *GetOrCreateSubComputationFromRegion(&unnested_reduce.body(), in EmitIRForReduction()