Searched refs:VariableOp (Results 1 – 3 of 3) sorted by relevance
50 VariableOp::VariableOp(OpKernelConstruction* context) : OpKernel(context) { in VariableOp() function in tensorflow::VariableOp55 void VariableOp::Compute(OpKernelContext* ctx) { in Compute()177 REGISTER_KERNEL_BUILDER(Name("Variable").Device(DEVICE_CPU), VariableOp);178 REGISTER_KERNEL_BUILDER(Name("VariableV2").Device(DEVICE_CPU), VariableOp);190 VariableOp); \193 VariableOp); \218 VariableOp); \221 VariableOp); \
31 class VariableOp : public OpKernel {33 explicit VariableOp(OpKernelConstruction* context);44 TF_DISALLOW_COPY_AND_ASSIGN(VariableOp);
610 friend class VariableOp; // For access to set_shape variable