Searched defs:buffer_to_size (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | tensor_list_ops_decomposition.cc | 74 llvm::SmallDenseMap<Value, SizeInfo>* buffer_to_size, in ModifyFunctionSignature()
118 Block& block, const llvm::SmallDenseMap<Value, SizeInfo>& buffer_to_size) { in AddTensorListSizesToTerminator()
142 const llvm::SmallDenseMap<Value, SizeInfo>& buffer_to_size) { in ModifyFunctionReturn()
151 llvm::SmallDenseMap<Value, SizeInfo>* buffer_to_size, in HandleWhileOp()
211 llvm::SmallDenseMap<Value, SizeInfo>* buffer_to_size, in HandleCaseOrIfOp()
269 llvm::SmallDenseMap<Value, SizeInfo>* buffer_to_size, in HandleWhileRegionOp()
334 llvm::SmallDenseMap<Value, SizeInfo>* buffer_to_size, in HandleIfRegionOp()
376 llvm::SmallDenseMap<Value, SizeInfo>* buffer_to_size, in HandleCaseRegionOp()
423 llvm::SmallDenseMap<Value, SizeInfo>* buffer_to_size, in HandlePartitionedCallOp()
554 llvm::SmallDenseMap<Value, SizeInfo>* buffer_to_size) { in HandleEmptyTensorListOp()
[all …]
|