| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | variable_ops.cc | 62 class AssignVariableOp : public XlaOpKernel { class
64 explicit AssignVariableOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in AssignVariableOp() function in tensorflow::__anond3063dcd0111::AssignVariableOp
70 REGISTER_XLA_OP(Name("AssignVariableOp"), AssignVariableOp);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | resource_variable_ops.cc | 210 class AssignVariableOp : public OpKernel { class
212 explicit AssignVariableOp(OpKernelConstruction* c) : OpKernel(c) { in AssignVariableOp() function in tensorflow::AssignVariableOp
339 class AssignVariableOp<Device, Variant> : public OpKernel { class
341 explicit AssignVariableOp(OpKernelConstruction* c) : OpKernel(c) { in AssignVariableOp() function in tensorflow::AssignVariableOp
387 AssignVariableOp<Eigen::ThreadPoolDevice, type>);
399 AssignVariableOp<GPUDevice, type>);
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_AssignVariableOp.pbtxt | 2 graph_op_name: "AssignVariableOp"
|
| /external/tensorflow/tensorflow/go/op/ |
| D | scope_test.go | 78 init = AssignVariableOp(s, variable, zero)
|
| D | wrappers.go | 13541 func AssignVariableOp(scope *Scope, resource tf.Output, value tf.Output) (o *tf.Operation) { func
|
| /external/tensorflow/tensorflow/core/grappler/costs/graph_properties_testdata/ |
| D | loops_and_resource_vars.pbtxt | 58 op: "AssignVariableOp"
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | single_machine_test.cc | 499 ops::AssignVariableOp(s.WithOpName("b/init/assign"), b, b_init); in CreateGrapplerItemWithResourceMemory()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor.h | 486 friend class AssignVariableOp; // For access to RefCountIsOne(). variable
|
| /external/tensorflow/tensorflow/compiler/tf2xla/g3doc/ |
| D | cpu_supported_ops.md | 21 `AssignVariableOp` | `dtype={bool,complex64,double,float,int32,int64,uint32,uint…
|
| D | gpu_supported_ops.md | 21 `AssignVariableOp` | `dtype={bool,complex64,double,float,int32,int64,uint32,uint…
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 2164 name: "AssignVariableOp"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v1.pbtxt | 6696 name: "AssignVariableOp"
|