Home
last modified time | relevance | path

Searched refs:buffer_arg_to_size_arg (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtensor_list_ops_decomposition.cc115 llvm::SmallDenseMap<int64_t, int64_t> buffer_arg_to_size_arg; member
446 auto arg_it = info.buffer_arg_to_size_arg.find(i); in HandlePartitionedCallOp()
447 if (arg_it == info.buffer_arg_to_size_arg.end()) continue; in HandlePartitionedCallOp()
518 info.buffer_arg_to_size_arg[buffer_arg.getArgNumber()] = in HandlePartitionedCallOp()