- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for 128x256xf32 (0.42 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/lite/tests/optimize_batch_matmul.mlir
%2 = "tfl.batch_matmul"(%1, %arg2) {adj_x = true, adj_y = false, asymmetric_quantize_inputs = false} : (tensor<4x8xf32>, tensor<4x256xf32>) -> tensor<8x256xf32> func.return %2 : tensor<8x256xf32> // CHECK: return %[[RES1]] : tensor<8x256xf32> } // CHECK-LABEL: Batchmatmul2Fullyconnected // CHECK-NOT: "tfl.batch_matmul"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 9K 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)