Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/api_def/java_api/
Dapi_def_AssignVariableOp.pbtxt2 graph_op_name: "AssignVariableOp"
/external/tensorflow/tensorflow/core/api_def/python_api/
Dapi_def_AssignVariableOp.pbtxt2 graph_op_name: "AssignVariableOp"
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dvariable_ops.cc81 class AssignVariableOp : public XlaOpKernel { class
83 explicit AssignVariableOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in AssignVariableOp() function in tensorflow::__anon5691afd60111::AssignVariableOp
89 REGISTER_XLA_OP(Name("AssignVariableOp").CompilationOnly(), AssignVariableOp);
/external/tensorflow/tensorflow/cc/tools/
Dfreeze_saved_model_test.cc133 auto assign = ops::AssignVariableOp(scope.WithOpName("assign"), var, a); in TestFreezeGraphWithoutDependentVariables()
178 auto assign = ops::AssignVariableOp(scope.WithOpName("assign"), var, a); in TestFreezeGraphWithDependentVariables()
229 auto assign = ops::AssignVariableOp(scope.WithOpName("assign"), var, a); in TestFreezeGraphWithAndWithoutDependentVariables()
236 ops::AssignVariableOp(scope.WithOpName("assign_1"), var_1, a); in TestFreezeGraphWithAndWithoutDependentVariables()
/external/tensorflow/tensorflow/core/kernels/
Dresource_variable_ops.cc320 class AssignVariableOp : public OpKernel { class
322 explicit AssignVariableOp(OpKernelConstruction* c) : OpKernel(c) { in AssignVariableOp() function in tensorflow::AssignVariableOp
387 class AssignVariableOp<Device, Variant> : public OpKernel { class
389 explicit AssignVariableOp(OpKernelConstruction* c) : OpKernel(c) { in AssignVariableOp() function in tensorflow::AssignVariableOp
468 AssignVariableOp<Eigen::ThreadPoolDevice, type>);
480 AssignVariableOp<GPUDevice, type>);
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_AssignVariableOp.pbtxt2 graph_op_name: "AssignVariableOp"
/external/tensorflow/tensorflow/go/op/
Dscope_test.go78 init = AssignVariableOp(s, variable, zero)
Dgradients_test.go225 init = AssignVariableOp(s, variable, zero)
Dwrappers.go26905 func AssignVariableOp(scope *Scope, resource tf.Output, value tf.Output) (o *tf.Operation) { func
/external/tensorflow/tensorflow/compiler/jit/
Dbuild_xla_ops_pass_test.cc111 ops::AssignVariableOp assign_op(scope.WithOpName("Assignee_" + id), in MakeWrite()
Dresource_operation_safety_analysis_test.cc55 ops::AssignVariableOp assign_op(scope.WithOpName("Assignee" + id), var_handle, in MakeWrite()
Dmark_for_compilation_pass_test.cc536 ops::AssignVariableOp assign_op(scope.WithOpName("Assignment" + id), in MakeWrite()
/external/tensorflow/tensorflow/core/common_runtime/
Dlower_if_op_test.cc185 auto init = ops::AssignVariableOp(root.WithOpName("init"), var, initial_val); in TEST()
/external/tensorflow/tensorflow/core/grappler/costs/graph_properties_testdata/
Dloops_and_resource_vars.pbtxt58 op: "AssignVariableOp"
/external/tensorflow/tensorflow/core/grappler/clusters/
Dsingle_machine_test.cc508 ops::AssignVariableOp(s.WithOpName("b/init/assign"), b, b_init); in CreateGrapplerItemWithResourceMemory()
/external/tensorflow/tensorflow/core/framework/
Dtensor.h621 friend class AssignVariableOp; // For access to RefCountIsOne(). variable
/external/tensorflow/tensorflow/compiler/tf2xla/g3doc/
Dcpu_supported_ops.md23 `AssignVariableOp` | `dtype={bool,complex64,double,float,int32,int64,uint32,uint…
Dgpu_supported_ops.md23 `AssignVariableOp` | `dtype={bool,complex64,double,float,int32,int64,uint32,uint…
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dloop_optimizer_test.cc1107 name: "EpisodicReplayBuffer/add/get_episode_id/critical_section_execute/AssignVariableOp/Switch" in TEST_F()
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.raw_ops.pbtxt208 name: "AssignVariableOp"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.raw_ops.pbtxt208 name: "AssignVariableOp"
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt2364 name: "AssignVariableOp"
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v2.pbtxt7359 name: "AssignVariableOp"
Dops_history.v1.pbtxt7359 name: "AssignVariableOp"