Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc5546 mlir::Operation* output_instruction = in EmitReductionFromOrToContiguousDimensions() local
5548 if (!IsReductionFromOrToContiguousDimensions(output_instruction)) { in EmitReductionFromOrToContiguousDimensions()
5557 auto reduce = mlir::cast<mlir::lmhlo::ReduceOp>(output_instruction); in EmitReductionFromOrToContiguousDimensions()