| /external/tensorflow/tensorflow/core/kernels/ |
| D | identity_op.cc | 26 REGISTER_KERNEL_BUILDER(Name("Identity").Device(DEVICE_CPU), IdentityOp);
27 REGISTER_KERNEL_BUILDER(Name("Identity").Device(DEVICE_TPU_SYSTEM), IdentityOp);
31 REGISTER_KERNEL_BUILDER(Name("StopGradient").Device(DEVICE_CPU), IdentityOp);
34 REGISTER_KERNEL_BUILDER(Name("PreventGradient").Device(DEVICE_CPU), IdentityOp);
39 IdentityOp);
41 REGISTER_KERNEL_BUILDER(Name("RefIdentity").Device(DEVICE_CPU), IdentityOp);
46 IdentityOp);
48 IdentityOp);
52 IdentityOp);
55 IdentityOp);
[all …]
|
| D | identity_op.h | 23 class IdentityOp : public OpKernel {
25 explicit IdentityOp(OpKernelConstruction* context) : OpKernel(context) {} in IdentityOp() function
|
| D | dynamic_partition_op_gpu.cu.cc | 119 struct IdentityOp { struct
128 : public TransformOutputIterator<int32, int32, IdentityOp> {
139 IdentityOp op, int32 limit) in BoundedReference()
154 BoundedOutputIterator(int32* __restrict__ ptr, IdentityOp op, int32 size) in BoundedOutputIterator()
159 IdentityOp op, int32 size) in BoundedOutputIterator()
404 IdentityOp id_op; in CountAndSortParts()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | identity_op.cc | 23 class IdentityOp : public XlaOpKernel { class
25 explicit IdentityOp(OpKernelConstruction* context) : XlaOpKernel(context) {} in IdentityOp() function in tensorflow::__anon3d5a6d050111::IdentityOp
42 TF_DISALLOW_COPY_AND_ASSIGN(IdentityOp);
49 IdentityOp);
54 IdentityOp);
55 REGISTER_XLA_OP(Name("PlaceholderWithDefault"), IdentityOp);
56 REGISTER_XLA_OP(Name("PreventGradient"), IdentityOp);
57 REGISTER_XLA_OP(Name("StopGradient").AllowVariantTypes(), IdentityOp);
58 REGISTER_XLA_OP(Name("Snapshot"), IdentityOp);
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | attribute_utils.h | 52 class IdentityOp; variable
61 if (!isa<IdentityOp>(op) && !isa<IdentityNOp>(op)) break; in GetValueAsConstant()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/quantization/tensorflow/ |
| D | tf_to_quant.cc | 88 if (auto id1 = dyn_cast_or_null<TF::IdentityOp>(min.getDefiningOp())) { in matchAndRewrite()
93 if (auto id2 = dyn_cast_or_null<TF::IdentityOp>(max.getDefiningOp())) { in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tensor_device_copy_conversion.cc | 44 func_op.walk([&](TF::IdentityOp op) { in runOnFunction()
|
| D | annotate_parameter_replication.cc | 51 if (!isa<TF::IdentityOp, TF::ReadVariableOp>(op)) break; in SkipIdentityAndReadVariable()
|
| D | tpu_identity_pruning.cc | 81 if (isa<TF::IdentityOp, TF::IdentityNOp>(op)) { in RemoveIdentityFromRegion()
|
| D | tpu_host_computation_expansion.cc | 55 return llvm::isa<TF::CastOp, TF::IdentityOp>(op); in IsTrivialUnaryOperation()
|
| D | tpu_sharding_identification_pass.cc | 94 if (llvm::isa<TF::IdentityOp, TF::CastOp, TF::ReadVariableOp>(owner)) { in GetXlaShardingFromArg()
225 if (llvm::isa_and_nonnull<TF::IdentityOp, TF::CastOp>(def)) { in GetXlaShardingFromRetval()
|
| D | tpu_colocate_composite_resource_ops.cc | 79 llvm::dyn_cast<TF::IdentityOp>(resource_user)) { in GetResourceOpsUsingCompositeArgsInReplicate()
|
| D | readonly_references_to_resources.cc | 119 if (!isa<IdentityOp>(user)) { in runOnFunction()
|
| D | tpu_device_propagation.cc | 104 isa<TF::IdentityOp, TF::IdentityNOp, TF::ShapeOp>(op); in IsSupportedOpToSetDevice()
|
| D | resource_device_inference.cc | 185 } else if (auto identity = dyn_cast<IdentityOp>(op)) { in ComputeResourceDevicesInComputation()
|
| D | tensor_list_ops_decomposition.cc | 621 Value buffer = builder.create<TF::IdentityOp>( in HandleTensorListFromTensorOp()
674 auto new_buffer = builder.create<TF::IdentityOp>( in HandleTensorListPopBackOp()
821 if (llvm::isa<TF::IdentityOp, TF::IdentityNOp>(&op)) { in DecomposeTensorListOpsInternal()
|
| D | collection_ops_util.cc | 199 } else if (llvm::isa<TF::IdentityOp, TF::IdentityNOp>(use.getOwner())) { in GetElementTypeFromAccess()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_test_util.h | 77 TFE_Op* IdentityOp(TFE_Context* ctx, TFE_TensorHandle* a);
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/analysis/ |
| D | side_effect_analysis.cc | 189 return llvm::isa<TF::IdentityNOp, TF::IdentityOp>(op) && in OpIsDeclaration()
226 if (isa<IdentityOp>(op)) return true; in OpIsKnownToHaveNoSideEffect()
|
| D | resource_alias_analysis.cc | 194 } else if (isa<IdentityNOp, IdentityOp>(op)) { in BacktrackValue()
312 } else if (llvm::isa<IdentityNOp, IdentityOp>(op)) { in ResourceAliasAnalysisInfo()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device_ops.h | 80 Name("Identity").Device(DEVICE).TypeConstraint("T", TYPES), IdentityOp); \
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | readonly_references_to_resources.mlir | 15 // Test case: Basic converting. '_class' attribute is at IdentityOp.
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | lower_static_tensor_list.cc | 784 struct ConvertIdentity : public OpConversionPattern<TF::IdentityOp> {
788 TF::IdentityOp op, ArrayRef<Value> operands, in matchAndRewrite()
791 rewriter.replaceOpWithNewOp<TF::IdentityOp>(op, input.getType(), operands, in matchAndRewrite()
|
| D | prepare_patterns.td | 63 // Partially supported in TFLite, treated as passthrough IdentityOp
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | breakup-islands.cc | 136 auto identity = builder.create<TF::IdentityOp>(island.getLoc(), in PopulateEmptyIsland()
|