Searched refs:DecomposeTensorArrayOps (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tensor_array_ops_decomposition.cc | 552 LogicalResult DecomposeTensorArrayOps(
579 if (failed(DecomposeTensorArrayOps(&body.front(), module, &body_stats, in HandleWhileOp()
581 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()
799 if (failed(DecomposeTensorArrayOps(&lowered_callee.front(), module, in HandlePartitionedCallOp()
854 if (failed(DecomposeTensorArrayOps(®ion.front(), module, stats, in HandleRegionControlFlowOps()
861 LogicalResult DecomposeTensorArrayOps( in DecomposeTensorArrayOps() function
941 if (failed(DecomposeTensorArrayOps(&main.front(), module, &stats, in runOnOperation()
|