Searched defs:data_var_to_size_var (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | stack_ops_decomposition.cc | 133 const llvm::SmallDenseMap<Value, Value>& data_var_to_size_var, in HandleWhileOp() argument
202 const llvm::SmallDenseMap<Value, Value>& data_var_to_size_var, in HandleIfOp() argument
259 const llvm::SmallDenseMap<Value, Value>& data_var_to_size_var, in HandlePartitionedCallOp() argument
353 llvm::SmallDenseMap<Value, Value>* data_var_to_size_var) { in HandleStackV2Op() argument
393 llvm::SmallDenseMap<Value, Value>* data_var_to_size_var) { in HandleStackPushV2Op() argument
420 llvm::SmallDenseMap<Value, Value>* data_var_to_size_var) { in HandleStackPopV2Op() argument
443 llvm::SmallDenseMap<Value, Value>* data_var_to_size_var, in HandleRegionControlFlowOps() argument
480 llvm::SmallDenseMap<Value, Value>* data_var_to_size_var, in DecomposeStackOpsInternal() argument
544 llvm::SmallDenseMap<Value, Value> data_var_to_size_var; in DecomposeStackOps() local
|