Home
last modified time | relevance | path

Searched defs:buffer_to_size (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtensor_list_ops_decomposition.cc74 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 …]