Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for scatter_dims_to_operand_dims (0.45 sec)

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

          SmallVector<int64_t, 4> permutation_array;
          for (int64_t i = 0; i < scatter_dims_to_operand_dims.size(); ++i) {
            permutation_array.push_back(scatter_dims_to_operand_dims[i]);
          }
          for (int64_t i = 0; i < operand_type.getRank(); ++i) {
            if (!llvm::is_contained(scatter_dims_to_operand_dims, i)) {
              permutation_array.push_back(i);
            }
          }
    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->inserted_window_dims, builder)));
        attributes.emplace_back(builder.getNamedAttr(
            "scatter_dims_to_operand_dims",
            BuildVhloTensorV1Attr(
                {static_cast<int64_t>(op->scatter_dims_to_operand_dims.size())},
                op->scatter_dims_to_operand_dims, builder)));
        attributes.emplace_back(builder.getNamedAttr(
    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

      %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>,
        indices_are_sorted = #vhlo.bool_v1<true>,
        unique_indices = #vhlo.bool_v1<true>}> ({
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Mar 14 19:15:40 UTC 2024
    - 31.9K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/schema/schema.fbs

      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-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
      // CHECK-NEXT: ^bb0(%[[ARG3:.*]]: tensor<f32>, %[[ARG4:.*]]: tensor<f32>):
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Apr 06 15:32:52 UTC 2024
    - 38.6K bytes
    - Viewed (0)
Back to top