- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for 512x256xf32 (0.1 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/rewrite_tpu_embedding_ops.mlir
"tf.Yield"(%0) : (tensor<512x256xf32>) -> () }, { "tf.Yield"(%arg2) : (tensor<512x256xf32>) -> () }) { is_stateless = true}: (tensor<i1>) -> tensor<512x256xf32> "tf.Yield"(%1) : (tensor<512x256xf32>) -> () }, { "tf.Yield"(%arg2) : (tensor<512x256xf32>) -> () }) { is_stateless = true}: (tensor<i1>) -> tensor<512x256xf32> func.return %2 : 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/tests/mlir2graphdef/tf_tpu_embedding_ops.mlir
%0:2 = tf_executor.island wraps "tf.RecvTPUEmbeddingActivations"() {config = "test_config_recv_embedding"} : () -> tensor<512x256xf32> loc("RecvTPUEmbedding") %1:1 = tf_executor.island wraps "tf.SendTPUEmbeddingGradients"(%0) {N = 1 : i64, NN = 0 : i64, config = "test_config_send_embedding", operandSegmentSizes = array<i32: 1, 0>} : (tensor<512x256xf32>) -> () loc("SendTPUEmbedding") tf_executor.fetch } func.return }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Aug 14 15:35:49 UTC 2023 - 1.9K bytes - Viewed (0)