- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for XlaRecvTPUEmbeddingDeduplicationData (0.55 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/rewrite_tpu_embedding_ops.mlir
// RUN: tf-opt -tf-rewrite-tpu-embedding-ops %s | FileCheck %s // CHECK-LABEL: func @recv_tpu_embedding_activations func.func @recv_tpu_embedding_activations() -> (tensor<512x256xf32>) { // CHECK: %[[DATA:.*]] = "tf.XlaRecvTPUEmbeddingDeduplicationData"() <{config = {{.*}}}> : () -> tensor<!tf_type.variant> // CHECK: %[[RESULT:.*]] = "tf.XlaRecvTPUEmbeddingActivations"(%[[DATA]]) <{config = {{.*}}}> : (tensor<!tf_type.variant>) -> tensor<512x256xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 4.2K bytes - Viewed (0) -
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)