Searched refs:IdentityNOp (Results 1 – 17 of 17) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | identity_n_op.cc | 27 REGISTER_KERNEL_BUILDER(Name("IdentityN").Device(DEVICE_DEFAULT), IdentityNOp);
29 IdentityNOp);
|
| D | identity_n_op.h | 23 class IdentityNOp : public OpKernel {
25 explicit IdentityNOp(OpKernelConstruction* context) : OpKernel(context) {} in IdentityNOp() function
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | attribute_utils.h | 53 class IdentityNOp; variable
61 if (!isa<IdentityOp>(op) && !isa<IdentityNOp>(op)) break; in GetValueAsConstant()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_identity_pruning.cc | 81 if (isa<TF::IdentityOp, TF::IdentityNOp>(op)) { in RemoveIdentityFromRegion()
|
| D | tpu_device_propagation.cc | 104 isa<TF::IdentityOp, TF::IdentityNOp, TF::ShapeOp>(op); in IsSupportedOpToSetDevice()
|
| D | collection_ops_util.cc | 199 } else if (llvm::isa<TF::IdentityOp, TF::IdentityNOp>(use.getOwner())) { in GetElementTypeFromAccess()
|
| D | resource_op_lifting_cleanup.cc | 45 if (isa<TF::IdentityOp, TF::IdentityNOp>(op) || IsCastOfResource(op)) { in RemovePassthroughOp()
|
| D | tpu_variable_runtime_reformatting.cc | 130 if (!llvm::isa<TF::IdentityOp, TF::IdentityNOp>(op)) { in SkipIdentity()
|
| D | tensor_array_ops_decomposition.cc | 486 if (llvm::isa<TF::IdentityOp, TF::IdentityNOp>(op)) { in AccessedGradients()
867 if (llvm::isa<TF::IdentityOp, TF::IdentityNOp>(&op)) { in DecomposeTensorArrayOps()
|
| D | stack_ops_decomposition.cc | 509 if (llvm::isa<TF::IdentityOp, TF::IdentityNOp>(&op)) { in DecomposeStackOpsInternal()
|
| D | tensor_list_ops_decomposition.cc | 821 if (llvm::isa<TF::IdentityOp, TF::IdentityNOp>(&op)) { in DecomposeTensorListOpsInternal()
|
| D | shape_inference.cc | 1039 if (isa<TF::IdentityOp, TF::IdentityNOp, TF::ZerosLikeOp>(op)) { in InferShapeForSingleOperation()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/analysis/ |
| D | resource_alias_analysis.cc | 194 } else if (isa<IdentityNOp, IdentityOp>(op)) { in BacktrackValue()
312 } else if (llvm::isa<IdentityNOp, IdentityOp>(op)) { in ResourceAliasAnalysisInfo()
|
| D | side_effect_analysis.cc | 189 return llvm::isa<TF::IdentityNOp, TF::IdentityOp>(op) && in OpIsDeclaration()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | breakup-islands.cc | 140 auto identity_n = builder.create<TF::IdentityNOp>( in PopulateEmptyIsland()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_common.cc | 358 .create<tosa::IdentityNOp>(op->getLoc(), ArrayRef<Type>(outs_type_vec), in convertUnpackOp()
1910 .create<tosa::IdentityNOp>(op->getLoc(), ArrayRef<Type>(outs_type_vec), in convertSplitOp()
1993 .create<tosa::IdentityNOp>(op->getLoc(), ArrayRef<Type>(outs_type_vec), in convertSplitVOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 1713 class ConvertIdentityNOp : public OpRewritePattern<TF::IdentityNOp> {
1715 using OpRewritePattern<TF::IdentityNOp>::OpRewritePattern;
1716 LogicalResult matchAndRewrite(TF::IdentityNOp op, in matchAndRewrite()
|