Home
last modified time | relevance | path

Searched defs:yieldOp (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
Dmerge_assuming_ops.cc127 auto yieldOp = llvm::cast<shape::AssumingYieldOp>(body->getTerminator()); in moveUpIntoAssumingOpMatchAndRewrite() local
290 auto yieldOp = cast<shape::AssumingYieldOp>(body->getTerminator()); in matchAndRewrite() local
Dconstraint_fusion_pass.cc319 auto yieldOp = in analyzeBroadcastableConstraint() local
385 auto yieldOp = in analyzeBlockGlobalConstraints() local
Drank_specialization.cc181 auto yieldOp = llvm::dyn_cast<chlo::RankSpecializationClusterYieldOp>( in matchAndRewrite() local
346 auto yieldOp = llvm::cast<chlo::RankSpecializationClusterYieldOp>( in materializeRankedOperations() local
355 auto yieldOp = llvm::cast<chlo::RankSpecializationClusterYieldOp>( in materializeFinalReshape() local
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/gml_st/transforms/
Dbufferizable_op_interface_impl.cc484 auto yieldOp = cast<SetYieldOp>(op); in getAliasingOpResult() local
494 auto yieldOp = cast<SetYieldOp>(op); in bufferize() local
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/gml_st/IR/
Dgml_st_ops.cc778 auto yieldOp = cast<YieldOp>(loopOp.getLoopBody().front().getTerminator()); in isShapePreserving() local
928 auto yieldOp = cast<YieldOp>(block->getTerminator()); in matchAndRewrite() local