Searched refs:ShapeOp (Results 1 – 7 of 7) sorted by relevance
29 ShapeOp<int32>);34 ShapeOp<int64>);43 ShapeOp<int32>); \49 ShapeOp<int64>);61 ShapeOp<int32>);68 ShapeOp<int64>);78 ShapeOp<int32>); \84 ShapeOp<int64>);100 ShapeOp<int32>);107 ShapeOp<int64>);
40 class ShapeOp : public OpKernel {42 explicit ShapeOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in ShapeOp() function
32 class ShapeOp : public XlaOpKernel { class34 explicit ShapeOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in ShapeOp() function in tensorflow::__anon812f91ee0111::ShapeOp49 REGISTER_XLA_OP(Name("Shape").CompilationOnly().IsMetadataOp(), ShapeOp);
47 TFE_Op* ShapeOp(TFE_Context* ctx, TFE_TensorHandle* a);
131 TFE_Op* ShapeOp(TFE_Context* ctx, TFE_TensorHandle* a) { in ShapeOp() function
842 TFE_Op* shape_op = ShapeOp(ctx, hgpu); in TEST()
138 ShapeOp<int32>); \144 ShapeOp<int64>); \