- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for indices_are_sorted (0.86 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/tests/legalize_hlo.mlir
%0 = "mhlo.gather"(%arg0, %arg1) { dimension_numbers = #mhlo.gather< collapsed_slice_dims = [0], index_vector_dim = 2, start_index_map = [0], >, indices_are_sorted = false, slice_sizes = dense<1> : tensor<1xi64> } : (tensor<147456xf16>, tensor<192x256x1xi32>) -> tensor<192x256xf16> func.return %0 : tensor<192x256xf16> }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 29 07:26:59 UTC 2024 - 340.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf.mlir
// CHECK-SAME: start_index_map = [0, 1] // CHECK-SAME: index_vector_dim = 1 // CHECK-SAME: indices_are_sorted = true // CHECK-SAME: slice_sizes = dense<[1, 1, 300]> : tensor<3xi64> %0 = "tf.XlaGather"(%arg0, %arg1, %cst) {dimension_numbers = "\0A\02\00\01\12\01\00\1A\02\00\01\20\01", indices_are_sorted = true} : (tensor<200x100x300xf32>, tensor<10x2xi32>, tensor<3xi64>) -> tensor<1x300x10xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 06 18:46:23 UTC 2024 - 335.5K bytes - Viewed (0)