Searched refs:ModifyFunctionSignature (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | stack_ops_decomposition.cc | 90 void ModifyFunctionSignature( in ModifyFunctionSignature() function
153 ModifyFunctionSignature(body, &body_map, find_arg_stack_type, in HandleWhileOp()
162 ModifyFunctionSignature(cond, nullptr, find_arg_stack_type); in HandleWhileOp()
215 ModifyFunctionSignature(then_func, &then_map, find_arg_stack_type); in HandleIfOp()
216 ModifyFunctionSignature(else_func, &else_map, find_arg_stack_type); in HandleIfOp()
320 ModifyFunctionSignature(lowered_callee, &callee_map, find_arg_stack_type); in HandlePartitionedCallOp()
|
| H A D | tensor_list_ops_decomposition.cc | 72 void ModifyFunctionSignature( in ModifyFunctionSignature() function
166 ModifyFunctionSignature(body, cutil::GetSizeType(builder), &body_map, in HandleWhileOp()
178 ModifyFunctionSignature(cond, cutil::GetSizeType(builder), &cond_map, in HandleWhileOp()
230 ModifyFunctionSignature(branch, cutil::GetSizeType(builder), &branch_map, in HandleCaseOrIfOp()
481 ModifyFunctionSignature(lowered_callee, cutil::GetSizeType(OpBuilder(call)), in HandlePartitionedCallOp()
|