Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 23 for indices_are_sorted (0.24 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. 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)
  6. 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)
  7. 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)
  8. 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)
  9. 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)
  10. 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)
Back to top