Searched defs:ShapeOp (Results 1 – 3 of 3) sorted by relevance
38 class ShapeOp : public XlaOpKernel { class40 explicit ShapeOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in ShapeOp() function in tensorflow::__anond5896c2a0111::ShapeOp
44 explicit ShapeOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in ShapeOp() function
293 TFE_Op* ShapeOp(TFE_Context* ctx, TFE_TensorHandle* a) { in ShapeOp() function