Home
last modified time | relevance | path

Searched defs:yield_op (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dbroadcast_propagation.cc113 auto yield_op = cast<shape::AssumingYieldOp>(body->getTerminator()); in MoveIntoAssumingOpMatchAndRewrite() local
225 auto yield_op = cast<shape::AssumingYieldOp>(body->getTerminator()); in matchAndRewrite() local
Drank_specialization.cc178 auto yield_op = llvm::dyn_cast<chlo::RankSpecializationClusterYieldOp>( in matchAndRewrite() local
340 auto yield_op = llvm::cast<chlo::RankSpecializationClusterYieldOp>( in MaterializeRankedOperations() local
349 auto yield_op = llvm::cast<chlo::RankSpecializationClusterYieldOp>( in MaterializeFinalReshape() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dexecutor_tpuv1_outline_tpu_island.cc125 YieldOp yield_op = island_op.GetYield(); in runOnOperation() local
Dexecutor_tpuv1_island_coarsening.cc263 YieldOp yield_op = island.GetYield(); in MergeIsland() local
Dresource_op_lifting_cleanup.cc380 Operation *yield_op = body.front().getTerminator(); in CanonicalizeWhileRegion() local
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dwhile_loop_outline.cc145 Operation* yield_op = outlined_func.getBody().front().getTerminator(); in CreateOutlineFunc() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_executor.cc326 YieldOp yield_op = op.GetYield(); in Print() local
987 YieldOp yield_op = island_op.GetYield(); in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dtf_to_jit_invocations.cc168 auto yield_op = in matchAndRewrite() local