Home
last modified time | relevance | path

Searched defs:IdentityOp (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Didentity_op.cc24 class IdentityOp : public XlaOpKernel { class
26 explicit IdentityOp(OpKernelConstruction* context) : XlaOpKernel(context) {} in IdentityOp() function in tensorflow::__anon547054b40111::IdentityOp
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Ddynamic_partition_op_gpu.cu.cc108 struct IdentityOp { struct
109 __device__ int32 __forceinline__ operator()(const int32& a) const { in operator ()()
H A Didentity_op.h25 explicit IdentityOp(OpKernelConstruction* context) : OpKernel(context) {} in IdentityOp() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
H A Dattribute_utils.h76 class IdentityOp; variable
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/
H A Dc_api_test_util.cc280 TFE_Op* IdentityOp(TFE_Context* ctx, TFE_TensorHandle* a) { in IdentityOp() function
/aosp_15_r20/external/swiftshader/third_party/llvm-16.0/llvm/lib/Transforms/Vectorize/
H A DSLPVectorizer.cpp6375 ShuffleVectorInst *IdentityOp = nullptr; in peekThroughShuffles() local
/aosp_15_r20/external/swiftshader/third_party/llvm-10.0/llvm/lib/Target/X86/
H A DX86ISelLowering.cpp36155 APInt IdentityOp = APInt::getAllOnesValue(NumOps); in SimplifyMultipleUseDemandedBitsForTargetNode() local
/aosp_15_r20/external/swiftshader/third_party/llvm-16.0/llvm/lib/Target/X86/
H A DX86ISelLowering.cpp43251 APInt IdentityOp = APInt::getAllOnes(NumOps); in SimplifyMultipleUseDemandedBitsForTargetNode() local