Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dresource_variable_ops.cc388 class AssignVariableOp : public OpKernel { class
390 explicit AssignVariableOp(OpKernelConstruction* c) : OpKernel(c) { in AssignVariableOp() function in tensorflow::AssignVariableOp
476 class AssignVariableOp<Device, Variant> : public OpKernel { class
478 explicit AssignVariableOp(OpKernelConstruction* c) : OpKernel(c) { in AssignVariableOp() function in tensorflow::AssignVariableOp
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dvariable_ops.cc96 class AssignVariableOp : public XlaOpKernel { class
98 explicit AssignVariableOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in AssignVariableOp() function in tensorflow::__anon735ed9830111::AssignVariableOp
/aosp_15_r20/external/tensorflow/tensorflow/c/experimental/ops/
H A Dresource_variable_ops.cc84 Status AssignVariableOp(AbstractContext* ctx, in AssignVariableOp() function
/aosp_15_r20/external/tensorflow/tensorflow/go/op/
H A Dwrappers.go1633 func AssignVariableOp(scope *Scope, resource tf.Output, value tf.Output, optional ...AssignVariable… func