Searched refs:XlaAssignVariableOp (Results 1 – 2 of 2) sorted by relevance
32 XlaAssignVariableOp::XlaAssignVariableOp(OpKernelConstruction* c) in XlaAssignVariableOp() function in tensorflow::XlaAssignVariableOp37 void XlaAssignVariableOp::Compute(OpKernelContext* context) { in Compute()
55 class XlaAssignVariableOp : public OpKernel {57 explicit XlaAssignVariableOp(OpKernelConstruction* c);175 XlaAssignVariableOp); \