Searched refs:VariableShapeOp (Results 1 – 2 of 2) sorted by relevance
42 class VariableShapeOp : public XlaOpKernel { class44 explicit VariableShapeOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in VariableShapeOp() function in tensorflow::__anon5691afd60111::VariableShapeOp61 REGISTER_XLA_OP(Name("VariableShape"), VariableShapeOp);
254 class VariableShapeOp : public OpKernel { class256 explicit VariableShapeOp(OpKernelConstruction* c) : OpKernel(c) {} in VariableShapeOp() function in tensorflow::VariableShapeOp276 VariableShapeOp<int32>);279 VariableShapeOp<int64>);288 VariableShapeOp<int32>);294 VariableShapeOp<int64>);