Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for inserted_window_dims (3.03 sec)

  1. tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/scatter.h

            return failure();
          }
    
          auto inserted_window_dims =
              scatter_dimension_numbers.getInsertedWindowDims();
          auto scatter_dims_to_operand_dims =
              scatter_dimension_numbers.getScatterDimsToOperandDims();
    
          if (IsIotaAttr(inserted_window_dims, indices_type.getShape().back()) &&
              IsIotaAttr(scatter_dims_to_operand_dims,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 10.1K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/flatbuffer_operator.cc

                op->update_window_dims, builder)));
        attributes.emplace_back(builder.getNamedAttr(
            "inserted_window_dims",
            BuildVhloTensorV1Attr(
                {static_cast<int64_t>(op->inserted_window_dims.size())},
                op->inserted_window_dims, builder)));
        attributes.emplace_back(builder.getNamedAttr(
            "scatter_dims_to_operand_dims",
            BuildVhloTensorV1Attr(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 18:21:50 UTC 2024
    - 38K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/vhlo.mlir

          tensor<200x100x300xf32> {
      %0 = "vhlo.scatter_v1" (%input_tensor, %scatter_indices, %updates) <{
        update_window_dims = #vhlo.tensor_v1<dense<1> : tensor<1xi64>>,
        inserted_window_dims = #vhlo.tensor_v1<dense<[0, 1]> : tensor<2xi64>>,
        scatter_dims_to_operand_dims = #vhlo.tensor_v1<dense<[0, 1]> : tensor<2xi64>>,
        index_vector_dim = #vhlo.integer_v1<1 : i64>,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Mar 14 19:15:40 UTC 2024
    - 31.9K bytes
    - Viewed (1)
  4. tensorflow/compiler/mlir/lite/schema/schema.fbs

      batch_group_count : long;
      precision_config : [StablehloPrecisionConfig];
    }
    
    table StablehloScatterOptions {
      indices_are_sorted: bool;
      update_window_dims: [long];
      inserted_window_dims: [long];
      scatter_dims_to_operand_dims: [long];
      index_vector_dim: long;
      unique_indices: bool;
      update_computation_subgraph_index: int;
    }
    
    enum RngAlgorithm : byte {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 03 18:01:23 UTC 2024
    - 41.7K bytes
    - Viewed (0)
  5. 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
      // CHECK-SAME: unique_indices = false
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Apr 06 15:32:52 UTC 2024
    - 38.6K bytes
    - Viewed (1)
Back to top