Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dpromote_resources_to_args.cc149 llvm::SmallVectorImpl<std::string>* var_handle_shared_names) { in PromoteVarHandlesToArguments() argument
163 var_handle_shared_names->emplace_back(name); in PromoteVarHandlesToArguments()
175 if (!var_handle_shared_names->empty()) in PromoteVarHandlesToArguments()
191 FuncOp function, llvm::ArrayRef<std::string> var_handle_shared_names) { in PromoteResourcesToArguments() argument
215 function.getNumArguments() - var_handle_shared_names.size()); in PromoteResourcesToArguments()
231 function.getArguments().take_back(var_handle_shared_names.size()); in PromoteResourcesToArguments()
289 function.getNumArguments() - var_handle_shared_names.size(); in PromoteResourcesToArguments()
309 var_handle_shared_names[index - var_handles_start_idx]}); in PromoteResourcesToArguments()
324 var_handle_shared_names[index - var_handles_start_idx])); in PromoteResourcesToArguments()
377 llvm::SmallVector<std::string, 4> var_handle_shared_names; in runOnOperation() local
[all …]