Searched defs:ShapeOp (Results 1 – 3 of 3) sorted by relevance
32 class ShapeOp : public XlaOpKernel { class34 explicit ShapeOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in ShapeOp() function in tensorflow::__anon812f91ee0111::ShapeOp
42 explicit ShapeOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in ShapeOp() function
131 TFE_Op* ShapeOp(TFE_Context* ctx, TFE_TensorHandle* a) { in ShapeOp() function