Searched refs:DecomposeTensorArrayOps (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | tensor_array_ops_decomposition.cc | 551 LogicalResult DecomposeTensorArrayOps(
578 if (failed(DecomposeTensorArrayOps(&body.front(), module, &body_stats, in HandleWhileOp()
580 failed(DecomposeTensorArrayOps(&cond.front(), module, &cond_stats, in HandleWhileOp()
663 if (failed(DecomposeTensorArrayOps(&then_branch.front(), module, &then_stats, in HandleIfOp()
665 failed(DecomposeTensorArrayOps(&else_branch.front(), module, &else_stats, in HandleIfOp()
800 if (failed(DecomposeTensorArrayOps(&lowered_callee.front(), module, in HandlePartitionedCallOp()
857 if (failed(DecomposeTensorArrayOps(®ion.front(), module, stats, in HandleRegionControlFlowOps()
864 LogicalResult DecomposeTensorArrayOps( in DecomposeTensorArrayOps() function
944 if (failed(DecomposeTensorArrayOps(&main.front(), module, &stats, in runOnOperation()
|