Search Options

Results per page
Sort
Preferred Languages
Advance

Results 21 - 30 of 38 for dimension_numbers (0.34 sec)

  1. tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_to_mhlo_int_test.cc

        %filter_scale, %filter_zp, %accum_scale, %accum_zp
      ) {
        Tin = "tfdtype$DT_QINT8", Tout = "tfdtype$DT_QINT32",
        attr_map = "", batch_group_count = 1 : i64,
        dimension_numbers = "\10\03\1A\02\01\02 \02(\032\02\00\01@\03J\02\01\02",
        explicit_padding = [], feature_group_count = 1 : i64, lhs_dilation = [1, 1],
        lhs_quantization_axis = -1 : i64, lhs_quantization_max_val = 127 : i64,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 03 01:03:21 UTC 2024
    - 35.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/stablehlo/transforms/tfl_stablehlo_pass.cc

                attrs.push_back(named_attr);
              }
    
              break;
            }
            case flexbuffers::FBT_VECTOR: {
              if (std::string{key} == "dimension_numbers") {
                auto value_vec = value.AsVector();
                auto vec1 = FlatbufferVecToMlirVec(value_vec[2].AsVector());
                auto vec2 = FlatbufferVecToMlirVec(value_vec[5].AsVector());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jan 24 06:08:43 UTC 2024
    - 10.8K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/quantization/stablehlo/ops/stablehlo_op_quant_spec_test.cc

            %0 = stablehlo.constant dense<1.0> : tensor<3x4x2xf32>
            %1 = stablehlo.constant dense<2> : tensor<2x3x2xi64>
            %2 = "stablehlo.gather"(%0, %1) {
              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>,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 04 07:19:09 UTC 2024
    - 14.8K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/quantization/tensorflow/tests/duplicate_shape_determining_constants.mlir

      // tf.XlaConvV2's 2, 3, 4, 5, 6 indices should be compile-time constants.
      %0 = "tf.XlaConvV2"(%arg0, %arg1, %strides, %padding, %lhs_dilation, %rhs_dilation, %feature_group_count) {
          batch_group_count = 1 : i64,
          dimension_numbers = "\18\03 \042\03\00\01\02@\04P\04Z\03\01\02\03b\03\01\02\03",
          precision_config = ""} : (tensor<8x4x16x16x16xf32>, tensor<4x3x3x16x16xf32>, tensor<3xi32>,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Nov 24 07:44:46 UTC 2022
    - 11K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/tests/tf-ops.mlir

      // / expected-error@+1 {{'tf.XlaConvV2' op expects feature_group_count to be a scalar}}
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Oct 23 14:40:35 UTC 2023
    - 236.4K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/stablehlo/tests/uniform-quantized-stablehlo-to-tfl.mlir

        %arg0: tensor<3x4x2x2x!quant.uniform<i8:f32, 3.000000e-01:-5>>,
        %arg1: tensor<2x3x2xi64>
      ) -> tensor<2x3x2x2x!quant.uniform<i8:f32, 3.000000e-01:-5>> {
      %0 = "stablehlo.gather"(%arg0, %arg1) {
        dimension_numbers = #stablehlo.gather<
          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>,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 14 17:10:32 UTC 2024
    - 106.2K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tf2xla/tests/legalize-tf.mlir

      // CHECK-DAG: %[[V42:.*]] = "mhlo.concatenate"(%[[V33]], %[[V32]]) <{dimension = 0 : i64}> : (tensor<1x22x128xi32>, tensor<1x22x128xi32>) -> tensor<2x22x128xi32>
      // CHECK-DAG: %[[V43:.*]] = "mhlo.gather"(%[[ARG]], %[[V42]]) <{dimension_numbers = #mhlo.gather<offset_dims = [0], collapsed_slice_dims = [1, 2], start_index_map = [1, 2]>, indices_are_sorted = false, slice_sizes = dense<[7, 1, 1]> : tensor<3xi64>}> : (tensor<7x140x128xi32>, tensor<2x22x128xi32>) -> tensor<7x22x128xi32>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 06 18:46:23 UTC 2024
    - 335.5K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_ops_to_mhlo.cc

      // ConvolutionDimensionNumbers.
      ::tensorflow::UniformQuantizedConvolutionDimensionNumbersAttr dnums_input;
      if (!dnums_input.ParseFromString(std::string(op.getDimensionNumbers()))) {
        return op->emitError("Parse dimension_numbers failed.");
      }
      xla::ConvolutionDimensionNumbers dnums =
          ConvertConvolutionDimensionNumbers(dnums_input);
    
      SmallVector<NamedAttribute> converted_attrs;
      for (auto attr : op->getAttrs()) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 30.9K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/lite/tests/prepare-tf.mlir

      %3 = "tf.Const"() {value = dense<1> : tensor<2xi32>} : () -> tensor<2xi32> loc("XlaConv/window_strides")
      %4 = "tf.XlaConvV2"(%arg0, %0, %3, %2, %3, %3, %1) {batch_group_count = 1 : i64, device = "", dimension_numbers = "\18\02 \032\02\00\01@\03P\03Z\02\01\02b\02\01\02", precision_config = ""} : (tensor<4x8x8x16xf32>, tensor<3x3x16x16xf32>, tensor<2xi32>, tensor<2x2xi32>, tensor<2xi32>, tensor<2xi32>, tensor<i32>) -> tensor<4x8x8x16xf32>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 29 07:26:59 UTC 2024
    - 59.8K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/quantization/tensorflow/tests/replace_cast_hacks_with_tf_xla_ops.mlir

    // CHECK: %[[conv_output:.*]] = "tf.XlaConvV2"(%[[input_padded]], %[[filter]], {{.*}}, {{.*}}, {{.*}}, {{.*}}, {{.*}}) <{batch_group_count = 1 : i64, dimension_numbers = "{{.*}}", precision_config = ""}> : (tensor<?x?x?x3xi8>, tensor<2x3x3x2xi8>, tensor<2xi32>, tensor<2x2xi32>, tensor<2xi32>, tensor<2xi32>, tensor<i32>) -> tensor<?x?x?x2xi32>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Oct 30 06:52:55 UTC 2023
    - 81K bytes
    - Viewed (0)
Back to top