Searched refs:HandleWhileOp (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | stack_ops_decomposition.cc | 159 LogicalResult HandleWhileOp( in HandleWhileOp() function
530 if (failed(HandleWhileOp(while_op, module, *data_var_to_size_var, in DecomposeStackOpsInternal()
|
| D | tensor_list_ops_decomposition.cc | 163 LogicalResult HandleWhileOp( in HandleWhileOp() function
889 if (failed(HandleWhileOp(while_op, module, buffer_to_size, in DecomposeTensorListOpsInternal()
|
| D | tensor_array_ops_decomposition.cc | 556 LogicalResult HandleWhileOp(TF::WhileOp while_op, ModuleOp module, in HandleWhileOp() function
895 if (failed(HandleWhileOp(while_op, module, stats, in DecomposeTensorArrayOps()
|