Searched refs:DecomposeStackOpsInternal (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | stack_ops_decomposition.cc | 125 LogicalResult DecomposeStackOpsInternal(
156 if (failed(DecomposeStackOpsInternal(&body.front(), module, &body_map, in HandleWhileOp()
164 if (failed(DecomposeStackOpsInternal(&cond.front(), module, &empty_map, in HandleWhileOp()
218 if (failed(DecomposeStackOpsInternal(&then_func.front(), module, &then_map, in HandleIfOp()
220 failed(DecomposeStackOpsInternal(&else_func.front(), module, &else_map, in HandleIfOp()
343 if (failed(DecomposeStackOpsInternal(&callee.front(), module, &callee_map, in HandlePartitionedCallOp()
465 if (failed(DecomposeStackOpsInternal(®ion.front(), module, in HandleRegionControlFlowOps()
478 LogicalResult DecomposeStackOpsInternal( in DecomposeStackOpsInternal() function
547 return DecomposeStackOpsInternal(block, module, &data_var_to_size_var, in DecomposeStackOps()
|