Searched defs:IdentityOp (Results 1 – 3 of 3) sorted by relevance
22 class IdentityOp : public XlaOpKernel { class24 explicit IdentityOp(OpKernelConstruction* context) : XlaOpKernel(context) {} in IdentityOp() function in tensorflow::__anonb0a82ca70111::IdentityOp
25 explicit IdentityOp(OpKernelConstruction* context) : OpKernel(context) {} in IdentityOp() function
113 struct IdentityOp { struct114 __device__ int32 __forceinline__ operator()(const int32& a) const { in operator ()()