- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for RewriteTPUEmbeddingOps (0.15 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/rewrite_tpu_embedding_ops.cc
// Rewrites RecvTPUEmbeddingActivationsOp and SendTPUEmbeddingGradients ops to // internal variants by introducing XlaRecvTPUEmbeddingDeduplicationData op. struct RewriteTPUEmbeddingOps : public impl::RewriteTPUEmbeddingOpsPassBase<RewriteTPUEmbeddingOps> { void runOnOperation() override; }; // Rewrites the given op to `OpT` op after adding the given operand at the end. template <typename OpT>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 15 22:55:42 UTC 2024 - 4.5K bytes - Viewed (0)