Searched refs:callee_map (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | stack_ops_decomposition.cc | 334 llvm::SmallDenseMap<Value, Value> callee_map; in HandlePartitionedCallOp() local
346 ModifyFunctionSignature(lowered_callee, &callee_map, find_arg_stack_type); in HandlePartitionedCallOp()
347 info.signature_change = !callee_map.empty(); in HandlePartitionedCallOp()
355 for (auto& entry : callee_map) { in HandlePartitionedCallOp()
368 if (failed(DecomposeStackOpsInternal(&callee.front(), module, &callee_map, in HandlePartitionedCallOp()
|
| D | tensor_list_ops_decomposition.cc | 478 llvm::SmallDenseMap<Value, SizeInfo> callee_map; in HandlePartitionedCallOp() local
494 &callee_map, find_arg_buffer_type, in HandlePartitionedCallOp()
496 const bool args_no_changed = callee_map.empty(); in HandlePartitionedCallOp()
498 &lowered_callee.front(), module, &callee_map, in HandlePartitionedCallOp()
503 ModifyFunctionReturn(lowered_callee, callee_map); in HandlePartitionedCallOp()
515 for (auto& entry : callee_map) { in HandlePartitionedCallOp()
|