Searched defs:IdentityOp (Results 1 – 8 of 8) sorted by relevance
24 class IdentityOp : public XlaOpKernel { class26 explicit IdentityOp(OpKernelConstruction* context) : XlaOpKernel(context) {} in IdentityOp() function in tensorflow::__anond1a3ff340111::IdentityOp
25 explicit IdentityOp(OpKernelConstruction* context) : OpKernel(context) {} in IdentityOp() function
108 struct IdentityOp { struct109 __device__ int32 __forceinline__ operator()(const int32& a) const { in operator ()()
76 class IdentityOp; variable
280 TFE_Op* IdentityOp(TFE_Context* ctx, TFE_TensorHandle* a) { in IdentityOp() function
6375 ShuffleVectorInst *IdentityOp = nullptr; in peekThroughShuffles() local
36155 APInt IdentityOp = APInt::getAllOnesValue(NumOps); in SimplifyMultipleUseDemandedBitsForTargetNode() local
43251 APInt IdentityOp = APInt::getAllOnes(NumOps); in SimplifyMultipleUseDemandedBitsForTargetNode() local