Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dpromote_resources_to_args.cc108 mlir::LogicalResult ValidateVarHandle(TF::VarHandleOp var_handle_op) { in ValidateVarHandle()
155 for (auto var_handle_op : in PromoteVarHandlesToArguments() local
Dreadonly_references_to_resources.cc148 VarHandleOp var_handle_op = builder.create<VarHandleOp>( in runOnFunction() local
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dvariable_merger_pass.cc178 Node* var_handle_op = nullptr; in Run() local
/external/tensorflow/tensorflow/compiler/jit/
Dmark_for_compilation_pass_test.cc653 Node** var_handle_op = nullptr) { in MakeRead()