Searched refs:body_map (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | stack_ops_decomposition.cc | 165 llvm::SmallDenseMap<Value, Value> body_map; in HandleWhileOp() local
180 ModifyFunctionSignature(body, &body_map, find_arg_stack_type, in HandleWhileOp()
182 const bool signature_change = !body_map.empty(); in HandleWhileOp()
183 if (failed(DecomposeStackOpsInternal(&body.front(), module, &body_map, in HandleWhileOp()
|
| D | tensor_list_ops_decomposition.cc | 170 llvm::SmallDenseMap<Value, SizeInfo> body_map; in HandleWhileOp() local
180 ModifyFunctionSignature(body, cutil::GetSizeType(builder), &body_map, in HandleWhileOp()
183 &body.front(), module, &body_map, in HandleWhileOp()
187 auto output_buffer_to_size = ModifyFunctionReturn(body, body_map); in HandleWhileOp()
|