Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dstack_ops_decomposition.cc161 const llvm::SmallDenseMap<Value, Value>& data_var_to_size_var, in HandleWhileOp() argument
167 auto it = data_var_to_size_var.find(while_op.getOperand(index)); in HandleWhileOp()
168 if (it == data_var_to_size_var.end()) return llvm::None; in HandleWhileOp()
201 auto it = data_var_to_size_var.find(while_op.getOperand(i)); in HandleWhileOp()
202 if (it == data_var_to_size_var.end()) continue; in HandleWhileOp()
229 const llvm::SmallDenseMap<Value, Value>& data_var_to_size_var, in HandleIfOp() argument
238 auto it = data_var_to_size_var.find(if_op.getOperand(index + 1)); in HandleIfOp()
239 if (it == data_var_to_size_var.end()) return llvm::None; in HandleIfOp()
254 auto it = data_var_to_size_var.find(operand); in HandleIfOp()
255 if (it == data_var_to_size_var.end()) continue; in HandleIfOp()
[all …]