- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 18 for indices_are_sorted (0.26 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/passes/lift_quantizable_spots_as_functions_simple.td
(LiftAsTFXlaCallModule<"composite_gather_fn"> (ArgumentList $operand, $start_indices), (ResultList $res), (NamedAttributeList (NamedAttr<"dimension_numbers"> $dimension_numbers), (NamedAttr<"slice_sizes"> $slice_sizes), (NamedAttr<"indices_are_sorted"> (DefaultOrNullAttr $indices_are_sorted)))),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 28 23:21:42 UTC 2024 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/tests/legalize-stablehlo-tfl-composite.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 5.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_operator.cc
op->unique_indices))); attributes.emplace_back(builder.getNamedAttr( "indices_are_sorted", mlir::vhlo::BooleanV1Attr::get(builder.getContext(), op->indices_are_sorted))); attributes.emplace_back(builder.getNamedAttr( "index_vector_dim", BuildVhloIntV1Attr(op->index_vector_dim, builder))); return; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 38K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/vhlo.mlir
start_index_map = #vhlo.tensor_v1<dense<[1, 0]> : tensor<2xi64>>, index_vector_dim = #vhlo.integer_v1<2 : i64>, slice_sizes = #vhlo.tensor_v1<dense<[1, 2, 2]> : tensor<3xi64>>, indices_are_sorted = #vhlo.bool_v1<false> }> : (tensor<3x4x2xi32>, tensor<2x3x2xi64>) -> tensor<2x3x2x2xi32> return %result : tensor<2x3x2x2xi32> }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 14 19:15:40 UTC 2024 - 31.9K bytes - Viewed (1) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_patterns.td
$dimension_numbers, $indices_are_sorted), (MHLO_GatherOp $operand, $start_indices, (ToGatherDimNumsAttr $dimension_numbers), (CastElementsToI64Elements $slice_sizes), $indices_are_sorted), [(HasValidGatherDims $dimension_numbers)]>;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 06 18:46:23 UTC 2024 - 34.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/ops/stablehlo_op_quant_spec_test.cc
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> func.return %2 : tensor<2x3x2x2xf32> } } )mlir";
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 04 07:19:09 UTC 2024 - 14.8K bytes - Viewed (0) -
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/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)