Searched refs:ShapeNOp (Results 1 – 4 of 4) sorted by relevance
116 ShapeNOp<int32>);121 ShapeNOp<int64>);130 ShapeNOp<int32>); \136 ShapeNOp<int64>)151 ShapeNOp<int32>);158 ShapeNOp<int64>);168 ShapeNOp<int32>); \174 ShapeNOp<int64>)186 ShapeNOp<int32>);193 ShapeNOp<int64>);
67 class ShapeNOp : public OpKernel {69 explicit ShapeNOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in ShapeNOp() function
51 class ShapeNOp : public XlaOpKernel { class53 explicit ShapeNOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in ShapeNOp() function in tensorflow::__anon812f91ee0111::ShapeNOp71 REGISTER_XLA_OP(Name("ShapeN").CompilationOnly().IsMetadataOp(), ShapeNOp);
150 ShapeNOp<int32>); \156 ShapeNOp<int64>); \