Searched defs: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()
133 Block& block, const llvm::SmallDenseMap<Value, SizeInfo>& buffer_to_size) { in AddTensorListSizesToTerminator()
156 FuncOp func, const llvm::SmallDenseMap<Value, SizeInfo>& buffer_to_size) { in ModifyFunctionReturn()
165 llvm::SmallDenseMap<Value, SizeInfo>* buffer_to_size, in HandleWhileOp()
225 llvm::SmallDenseMap<Value, SizeInfo>* buffer_to_size, in HandleCaseOrIfOp()
283 llvm::SmallDenseMap<Value, SizeInfo>* buffer_to_size, in HandleWhileRegionOp()
347 llvm::SmallDenseMap<Value, SizeInfo>* buffer_to_size, in HandleIfRegionOp()
389 llvm::SmallDenseMap<Value, SizeInfo>* buffer_to_size, in HandleCaseRegionOp()
436 llvm::SmallDenseMap<Value, SizeInfo>* buffer_to_size, in HandlePartitionedCallOp()
564 llvm::SmallDenseMap<Value, SizeInfo>* buffer_to_size) { in HandleEmptyTensorListOp()
[all …]
|