Searched refs:buffer_to_size (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tensor_list_ops_decomposition.cc | 90 llvm::SmallDenseMap<Value, SizeInfo>* buffer_to_size, in ModifyFunctionSignature() argument
102 if (buffer_to_size) { in ModifyFunctionSignature()
103 (*buffer_to_size)[func.getArgument(i)] = {size_arg, in ModifyFunctionSignature()
133 Block& block, const llvm::SmallDenseMap<Value, SizeInfo>& buffer_to_size) { in AddTensorListSizesToTerminator() argument
139 auto it = buffer_to_size.find(retval.value()); in AddTensorListSizesToTerminator()
140 if (it == buffer_to_size.end()) continue; in AddTensorListSizesToTerminator()
156 FuncOp func, const llvm::SmallDenseMap<Value, SizeInfo>& buffer_to_size) { in ModifyFunctionReturn() argument
158 AddTensorListSizesToTerminator<ReturnOp>(func.front(), buffer_to_size); in ModifyFunctionReturn()
165 llvm::SmallDenseMap<Value, SizeInfo>* buffer_to_size, in HandleWhileOp() argument
172 auto it = buffer_to_size->find(while_op.getOperand(index)); in HandleWhileOp()
[all …]
|