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 | 74 llvm::SmallDenseMap<Value, SizeInfo>* buffer_to_size, in ModifyFunctionSignature() 117 Block& block, const llvm::SmallDenseMap<Value, SizeInfo>& buffer_to_size) { in AddTensorListSizesToTerminator() 140 FuncOp func, const llvm::SmallDenseMap<Value, SizeInfo>& buffer_to_size) { in ModifyFunctionReturn() 149 llvm::SmallDenseMap<Value, SizeInfo>* buffer_to_size, in HandleWhileOp() 209 llvm::SmallDenseMap<Value, SizeInfo>* buffer_to_size, in HandleCaseOrIfOp() 267 llvm::SmallDenseMap<Value, SizeInfo>* buffer_to_size, in HandleWhileRegionOp() 331 llvm::SmallDenseMap<Value, SizeInfo>* buffer_to_size, in HandleIfRegionOp() 373 llvm::SmallDenseMap<Value, SizeInfo>* buffer_to_size, in HandleCaseRegionOp() 420 llvm::SmallDenseMap<Value, SizeInfo>* buffer_to_size, in HandlePartitionedCallOp() 548 llvm::SmallDenseMap<Value, SizeInfo>* buffer_to_size) { in HandleEmptyTensorListOp() [all …]
|