- Sort Score
- Result 10 results
- Languages All
Results 41 - 50 of 79 for identity_op (0.31 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_variable_runtime_reformatting.cc
while (auto result = mlir::dyn_cast<OpResult>(v)) { if (!(allow_other_use || v.hasOneUse())) break; auto op = result.getDefiningOp(); if (!llvm::isa<TF::IdentityOp, TF::IdentityNOp>(op)) { break; } v = op->getOperand(result.getResultNumber()); if (skipped) skipped->insert(op); } return v; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalization_op_config.cc
TypeID::get<TF::_FusedConv2DOp>(), TypeID::get<TF::GatherNdOp>(), TypeID::get<TF::GatherV2Op>(), TypeID::get<TF::GreaterEqualOp>(), TypeID::get<TF::IdentityOp>(), TypeID::get<TF::IdentityNOp>(), TypeID::get<TF::InplaceUpdateOp>(), TypeID::get<TF::InvertPermutationOp>(), TypeID::get<TF::IRFFTOp>(), TypeID::get<TF::L2LossOp>(), TypeID::get<TF::LegacyCallOp>(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 04:08:35 UTC 2024 - 21.7K bytes - Viewed (0) -
platforms/software/publish/src/main/java/org/gradle/api/publish/internal/metadata/ModuleMetadataSpecBuilder.java
return ownerData != null ? identityFor(ownerData, relativeUrlTo(componentData.coordinates, ownerData.coordinates)) : identityFor(componentData, null); } private ModuleMetadataSpec.Identity identityFor(ComponentData componentData, String relativeUrl) { return new ModuleMetadataSpec.Identity(
Registered: Wed Jun 12 18:38:38 UTC 2024 - Last Modified: Tue Mar 05 02:07:04 UTC 2024 - 23K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/integration/node_expansion_pass.cc
// isn't a composite op. The following ops are explicitly skipped here because // their "no-op" expansion is known to cause problems in some cases. static const char* kOpsToSkip[] = { "IdentityOp", "NoOp", // b/174596063 "OptionalHasValue", // b/173136483 "OptionalGetValue", // b/173136483 "VarHandleOp", // b/176819198 }; for (const char* skip : kOpsToSkip) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Feb 25 16:22:36 UTC 2024 - 3.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_host_computation_expansion.cc
return value.getDefiningOp(); } // TODO(b/158596585): Replace this with a cost model analysis. bool IsTrivialUnaryOperation(Operation* op) { return llvm::isa<TF::CastOp, TF::IdentityOp>(op); } // Adds outside compilation attributes to unary ops such as Identity/Cast ops // at the head of TPU computation that is used only by other outside compiled
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
DatasetInput GetDatasetInput(Value value) { // TODO(haoliang): add an interface for DatasetOp to avoid the following // enumeration. // Iteratively tracing upwards if parent op is `IdentityOp` or `IdentityNOp`. while ( llvm::isa_and_nonnull<IdentityOp, IdentityNOp>(value.getDefiningOp())) { value = value.getDefiningOp()->getOperand( mlir::cast<OpResult>(value).getResultNumber()); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_colocate_composite_resource_ops.cc
for (auto resource_user : resource_users_to_visit) { add_resource_op_to_list(resource_user); // Account for pass-through identity ops. if (auto pass_through_identity = llvm::dyn_cast<TF::IdentityOp>(resource_user)) { for (auto identity_user : pass_through_identity.getOutput().getUsers()) { new_resource_users.emplace_back(identity_user); } } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 16 17:41:12 UTC 2023 - 5.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/stack_ops_decomposition.cc
llvm::StringMap<PartitionedCallStackOpsInfo>* decomposed_partitioned_call_callees) { for (auto& op : llvm::make_early_inc_range(block->getOperations())) { if (llvm::isa<TF::IdentityOp, TF::IdentityNOp>(&op)) { // Removes identity nodes in the block. The device computation does not // need such nodes to carry information. op.replaceAllUsesWith(op.getOperands()); op.erase();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 23.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/split_into_island_per_op.mlir
func.return } // CHECK-LABEL: empty_island_multiple_data_results // CHECK-SAME: (%[[ARG_0:.*]]: tensor<*xf32>, %[[ARG_1:.*]]: tensor<*xi32>) // CHECK: %[[IDENTITY_N:.*]]:2 = "tf.IdentityN" // CHECK-SAME: (%[[ARG_0]], %[[ARG_1]]) // CHECK: tf_executor.yield %[[IDENTITY_N]]#0, %[[IDENTITY_N]]#1 func.func @single_op_island_forward_block_arg(%arg0: tensor<?x?x?x?xbf16>) -> (tensor<2048xf32>, tensor<?x?x?x?xbf16>) { %0:2 = tf_executor.graph {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 20.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/merge_save_function_ops_to_main.mlir
} } // ----- // Test that the @tf_quant_save's ops are cloned to @main. When there are no // __tf_file_prefix argument in @main, confirm that it is created and wired // to the newly created `IdentityOp`. module attributes {tf_saved_model.semantics} { func.func private @tf_quant__save(%arg: tensor<!tf_type.string>) -> () { tf_executor.graph {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 7.7K bytes - Viewed (0)