- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for XlaRecvTPUEmbeddingDeduplicationData (0.25 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/rewrite_tpu_embedding_ops.cc
#include "tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.h.inc" // Rewrites RecvTPUEmbeddingActivationsOp and SendTPUEmbeddingGradients ops to // internal variants by introducing XlaRecvTPUEmbeddingDeduplicationData op. struct RewriteTPUEmbeddingOps : public impl::RewriteTPUEmbeddingOpsPassBase<RewriteTPUEmbeddingOps> { void runOnOperation() override; };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 15 22:55:42 UTC 2024 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td
embedding table in the model.}]>:$outputs ); TF_DerivedResultSizeAttr num_tables = TF_DerivedResultSizeAttr<0>; } def TF_XlaRecvTPUEmbeddingDeduplicationDataOp : TF_Op<"XlaRecvTPUEmbeddingDeduplicationData", []> { let summary = [{ Receives deduplication data (indices and weights) from the embedding core. }]; let description = [{
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K bytes - Viewed (0)