- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 20 for IdentityNOp (0.15 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/breakup-islands.cc
operand.getType(), operand); yield.setOperand(0, identity.getOutput()); } else { auto identity_n = builder.create<TF::IdentityNOp>( island.getLoc(), yield.getOperandTypes(), yield.getOperands()); for (auto it : llvm::enumerate(identity_n.getResults())) yield.setOperand(it.index(), it.value()); } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Aug 11 20:52:36 UTC 2023 - 16.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tensor_array_ops_decomposition.cc
// Walk all operations and nested regions to find accessed gradient sources // for function arguments. func.walk([&](Operation* op) { if (llvm::isa<TF::IdentityOp, TF::IdentityNOp>(op)) { op->replaceAllUsesWith(op->getOperands()); return; } if (auto grad = llvm::dyn_cast<TF::TensorArrayGradV3Op>(op)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 02 20:41:19 UTC 2023 - 40.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting_cleanup.cc
// such nodes to carry information. void RemovePassthroughOp(Block &block) { for (auto &op : llvm::make_early_inc_range(block)) { if (isa<TF::IdentityOp, TF::IdentityNOp>(op) || IsCastOfResource(op)) { op.replaceAllUsesWith(op.getOperands()); op.erase(); } } } using LocalVarOp = std::variant<TF::VarHandleOp, TF::MlirLocalVarOp>;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 19.7K bytes - Viewed (0) -
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) -
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/transforms/tensor_list_ops_decomposition.cc
for (auto& op : llvm::make_early_inc_range(block->getOperations())) { // TODO(yuanzx): Add a pass to remove identities in device computation. if (llvm::isa<TF::IdentityOp, TF::IdentityNOp, TF::StopGradientOp>(&op)) { op.replaceAllUsesWith(op.getOperands()); op.erase(); } else if (auto list = llvm::dyn_cast<TF::EmptyTensorListOp>(&op)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 39.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
// 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()); } Operation* op = value.getDefiningOp();
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/tf2xla/transforms/legalize_tf.cc
} return success(); } }; // Bypasses IdentityN op. class ConvertIdentityNOp : public OpRewritePattern<TF::IdentityNOp> { public: using OpRewritePattern<TF::IdentityNOp>::OpRewritePattern; LogicalResult matchAndRewrite(TF::IdentityNOp op, PatternRewriter &rewriter) const override { rewriter.replaceOp(op, op.getOperands()); return success();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
namespace mlir { namespace TF { namespace { // Returns the equivalent Value skipping through identity nodes. Value LookThroughIdentity(Value result) { while (isa_and_nonnull<IdentityOp, IdentityNOp>(result.getDefiningOp())) { auto op_result = result.cast<OpResult>(); result = op_result.getOwner()->getOperand(op_result.getResultNumber()); } return result; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0)