- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 23 for indices_are_sorted (0.24 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/tests/quantize_weights.mlir
%cst = "tf.Const"() { value = dense<[1, 1, 300]> : tensor<3xi64> } : () -> tensor<3xi64> %0 = "tf.XlaGather"(%cst_0, %arg0, %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> %1 = "tf.Identity"(%0) {device = ""} : (tensor<1x300x10xf32>) -> tensor<1x300x10xf32> func.return %1 : tensor<1x300x10xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 42K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/schema/schema.fbs
} table StablehloGatherOptions{ offset_dims : [long]; collapsed_slice_dims : [long]; start_index_map : [long]; index_vector_dim : long; slice_sizes : [long]; indices_are_sorted : bool; } table StablehloTransposeOptions{ permutation : [long]; } enum StablehloPrecisionConfig : uint { DEFAULT, HIGH, HIGHEST, }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:01:23 UTC 2024 - 41.7K 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) -
tensorflow/compiler/mlir/lite/stablehlo/tests/uniform-quantized-stablehlo-to-tfl.mlir
offset_dims = [2, 3], collapsed_slice_dims = [0, 1], start_index_map = [0, 1], index_vector_dim = 2>, slice_sizes = array<i64: 1, 1, 2, 2>, indices_are_sorted = false } : ( tensor<3x4x2x2x!quant.uniform<i8:f32, 3.000000e-01:-5>>, tensor<2x3x2xi64> ) -> tensor<2x3x2x2x!quant.uniform<i8:f32, 3.000000e-01:-5>>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 17:10:32 UTC 2024 - 106.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/schema/schema_generated.h
auto _indices_are_sorted = _o->indices_are_sorted; return tflite::CreateStablehloGatherOptions( _fbb, _offset_dims, _collapsed_slice_dims, _start_index_map, _index_vector_dim, _slice_sizes, _indices_are_sorted); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 1M bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_export.cc
scatter_dims_to_operand_dims_mlir.end()); int64_t index_vector_dim = scatter_dimension_numbers.getIndexVectorDim(); bool unique_indices = scatter_op.getUniqueIndices(); bool indices_are_sorted = scatter_op.getIndicesAreSorted(); auto update_window_dims = builder_.CreateVector(update_window_dims_vec); auto inserted_window_dims = builder_.CreateVector(inserted_window_dims_vec);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (2) -
tensorflow/compiler/mlir/quantization/stablehlo/tests/passes/quantize/quantize_same_scale.mlir
dimension_numbers = #stablehlo.gather< offset_dims = [2, 3], collapsed_slice_dims = [0], start_index_map = [1, 0], index_vector_dim = 2>, slice_sizes = array<i64: 1, 2, 2>, indices_are_sorted = false } : (tensor<3x4x2xf32>, tensor<2x3x2xi64>) -> tensor<2x3x2x2xf32> %8 = "quantfork.qcast"(%7) {volatile} : (tensor<2x3x2x2xf32>) -> tensor<2x3x2x2x!quant.uniform<i8:f32, 0.13170163023705575:-1>>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 17:10:32 UTC 2024 - 35.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf-with-tf2xla-hlo-importer.mlir
// CHECK: %[[RESULT:.*]] = "mhlo.scatter"(%[[DEFAULT]], %[[ARG0]], %[[ARG1]]) // CHECK-SAME: indices_are_sorted = false // CHECK-SAME: scatter_dimension_numbers // CHECK-SAME: inserted_window_dims = [0, 1] // CHECK-SAME: scatter_dims_to_operand_dims = [0, 1] // CHECK-SAME: index_vector_dim = 1
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 06 15:32:52 UTC 2024 - 38.6K bytes - Viewed (1) -
tensorflow/compiler/mlir/quantization/stablehlo/tests/passes/quantize_composite_functions.mlir
offset_dims = [2, 3], collapsed_slice_dims = [0], start_index_map = [1, 0], index_vector_dim = 2>, slice_sizes = array<i64: 1, 2, 2>, indices_are_sorted = false } : (tensor<3x4x2xf32>, tensor<2x3x2xi32>) -> tensor<2x3x2x2xf32> return %0 : tensor<2x3x2x2xf32> }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 05:56:10 UTC 2024 - 91.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/tests/passes/lift_quantizable_spots_as_functions.mlir
dimension_numbers = #stablehlo.gather< offset_dims = [2, 3], collapsed_slice_dims = [0], start_index_map = [1, 0], index_vector_dim = 2>, slice_sizes = array<i64: 1, 2, 2>, indices_are_sorted = false } : (tensor<3x4x2xi32>, tensor<2x3x2xi64>) -> tensor<2x3x2x2xi32> func.return %2: tensor<2x3x2x2xi32> } // CHECK: %[[OPERAND:.*]] = stablehlo.constant // CHECK: %[[INDICES:.*]] = stablehlo.constant
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 04:07:09 UTC 2024 - 49.8K bytes - Viewed (0)