- Sort Score
- Result 10 results
- Languages All
Results 11 - 11 of 11 for 1x10xi64 (0.11 sec)
-
tensorflow/compiler/mlir/lite/tests/ops.mlir
// CHECK-LABEL: testMultinomial func.func @testMultinomial(%arg0: tensor<2xf32>, %arg1: tensor<1xi32>) -> tensor<10xi64> { // CHECK: "tfl.multinomial"(%arg0, %arg1) %0 = "tfl.multinomial"(%arg0, %arg1) {seed = 0 : i64, seed2 = 0: i64} : (tensor<2xf32>, tensor<1xi32>) -> tensor<10xi64> func.return %0 : tensor<10xi64> } // ----- // CHECK-LABEL: testMultinomialInt32
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 19:09:08 UTC 2024 - 189.2K bytes - Viewed (0)