Searched defs:IdentityOp (Results 1 – 4 of 4) sorted by relevance
22 class IdentityOp : public XlaOpKernel { class24 explicit IdentityOp(OpKernelConstruction* context) : XlaOpKernel(context) {} in IdentityOp() function in tensorflow::__anonf9fce7290111::IdentityOp
25 explicit IdentityOp(OpKernelConstruction* context) : OpKernel(context) {} in IdentityOp() function
123 struct IdentityOp { struct124 __device__ int32 __forceinline__ operator()(const int32& a) const { in operator ()()
149 TFE_Op* IdentityOp(TFE_Context* ctx, TFE_TensorHandle* a) { in IdentityOp() function