Searched refs:ModifyFunctionSignature (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | stack_ops_decomposition.cc | 118 void ModifyFunctionSignature( in ModifyFunctionSignature() function
180 ModifyFunctionSignature(body, &body_map, find_arg_stack_type, in HandleWhileOp()
189 ModifyFunctionSignature(cond, nullptr, find_arg_stack_type); in HandleWhileOp()
242 ModifyFunctionSignature(then_func, &then_map, find_arg_stack_type); in HandleIfOp()
243 ModifyFunctionSignature(else_func, &else_map, find_arg_stack_type); in HandleIfOp()
346 ModifyFunctionSignature(lowered_callee, &callee_map, find_arg_stack_type); in HandlePartitionedCallOp()
|
| D | tensor_list_ops_decomposition.cc | 88 void ModifyFunctionSignature( in ModifyFunctionSignature() function
180 ModifyFunctionSignature(body, cutil::GetSizeType(builder), &body_map, in HandleWhileOp()
192 ModifyFunctionSignature(cond, cutil::GetSizeType(builder), &cond_map, in HandleWhileOp()
244 ModifyFunctionSignature(branch, cutil::GetSizeType(builder), &branch_map, in HandleCaseOrIfOp()
493 ModifyFunctionSignature(lowered_callee, cutil::GetSizeType(OpBuilder(call)), in HandlePartitionedCallOp()
|