Searched refs:DecomposeStackOpsInternal (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | stack_ops_decomposition.cc | 153 LogicalResult DecomposeStackOpsInternal(
183 if (failed(DecomposeStackOpsInternal(&body.front(), module, &body_map, in HandleWhileOp()
191 if (failed(DecomposeStackOpsInternal(&cond.front(), module, &empty_map, in HandleWhileOp()
245 if (failed(DecomposeStackOpsInternal(&then_func.front(), module, &then_map, in HandleIfOp()
247 failed(DecomposeStackOpsInternal(&else_func.front(), module, &else_map, in HandleIfOp()
368 if (failed(DecomposeStackOpsInternal(&callee.front(), module, &callee_map, in HandlePartitionedCallOp()
490 if (failed(DecomposeStackOpsInternal(®ion.front(), module, in HandleRegionControlFlowOps()
503 LogicalResult DecomposeStackOpsInternal( in DecomposeStackOpsInternal() function
572 return DecomposeStackOpsInternal(block, module, &data_var_to_size_var, in DecomposeStackOps()
|