Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dlower_static_tensor_list.cc166 FuncOp func, const llvm::SmallSet<int, 4> &tensor_list_args) { in GetResizedTensorListIndexes()
1102 ArrayRef<Value> operands, const llvm::SmallSet<int, 4> &tensor_list_args, in UpdateFunctionTypesForWhileOp()
1148 const llvm::SmallSet<int, 4> &tensor_list_args, in UpdateFunctionTypesForIfOp()
1265 auto tensor_list_args = GetTensorListArgumentsIndex(op.else_function()); in matchAndRewrite() local
1302 auto tensor_list_args = GetTensorListArgumentsIndex(op.body_function()); in matchAndRewrite() local