- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for identity_op (0.14 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_pipelining.cc
while (true) { if (auto block_arg = llvm::dyn_cast<BlockArgument>(val)) { return block_arg.getArgNumber(); } if (auto identity_op = llvm::dyn_cast<TF::IdentityOp>(val.getDefiningOp())) { val = identity_op.getOperand(); } else { return -1; } } }; for (auto call_op :
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 92.9K bytes - Viewed (0) -
.bazelrc
# https://github.com/tensorflow/tensorflow/issues/48919. # Users can still include debug info for a specific kernel, e.g. with: # --config=dbg --per_file_copt=+tensorflow/core/kernels/identity_op.*@-g # Since this .bazelrc file is synced between the tensorflow/tensorflow repo and # the openxla/xla repo, also include debug info for files under xla/. build:dbg --per_file_copt=+.*,-tensorflow.*,-xla.*@-g0
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 17:12:54 UTC 2024 - 52.9K bytes - Viewed (0) -
tensorflow/c/eager/c_api_test.cc
TFE_Op* identityOp = TFE_NewOp(ctx, "IdentityN", status); CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); // Try to retrieve lengths before building the attributes (should fail) EXPECT_EQ(-1, TFE_OpGetInputLength(identityOp, "input", status)); CHECK_NE(TF_OK, TF_GetCode(status)) << TF_Message(status); EXPECT_EQ(-1, TFE_OpGetOutputLength(identityOp, "output", status));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 03 20:50:20 UTC 2023 - 94.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc
} }; struct ConvertIdentity : public OpConversionPattern<TF::IdentityOp> { using OpConversionPattern::OpConversionPattern; LogicalResult matchAndRewrite( TF::IdentityOp op, OpAdaptor adaptor, ConversionPatternRewriter &rewriter) const override { Value input = adaptor.getOperands()[0]; rewriter.replaceOpWithNewOp<TF::IdentityOp>( op, input.getType(), adaptor.getOperands(), op->getAttrs());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 70.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc
// of Identity op is only used by TF ops. struct RemoveIdentity : public OpRewritePattern<TF::IdentityOp> { using OpRewritePattern<TF::IdentityOp>::OpRewritePattern; LogicalResult matchAndRewrite(TF::IdentityOp identity, PatternRewriter &rewriter) const override { // Replace the op with the input if input and result have the same type.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 64.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/g3doc/_includes/tf_passes.md
tf_executor.yield %identity : tensor<i32> } %island_1_result, %island_1_control = tf_executor.island { %identity_n:2 = "tf.IdentityN"(%arg1, %island_0_result) : (tensor<i32>, tensor<i32>) -> (tensor<i32>, tensor<i32>) tf_executor.yield %identity_n#0 } tf_executor.fetch %island_0_result, %island_1_result : tensor<i32>, tensor<i32> }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 02 02:26:39 UTC 2023 - 96.4K bytes - Viewed (0)