Searched refs:DecomposeTensorArrayOps (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| 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()
662 if (failed(DecomposeTensorArrayOps(&then_branch.front(), module, &then_stats, in HandleIfOp()
664 failed(DecomposeTensorArrayOps(&else_branch.front(), module, &else_stats, in HandleIfOp()
798 if (failed(DecomposeTensorArrayOps(&lowered_callee.front(), module, in HandlePartitionedCallOp()
853 if (failed(DecomposeTensorArrayOps(®ion.front(), module, stats, in HandleRegionControlFlowOps()
860 LogicalResult DecomposeTensorArrayOps( in DecomposeTensorArrayOps() function
940 if (failed(DecomposeTensorArrayOps(&main.front(), module, &stats, in runOnOperation()
|