- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 11 for scatter_dims_to_operand_dims (0.73 sec)
-
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) -
tensorflow/compiler/mlir/lite/stablehlo/tests/legalize-stablehlo-tfl-composite.mlir
stablehlo.return %arg6 : tensor<f32> }) {indices_are_sorted = false, scatter_dimension_numbers = #stablehlo.scatter<update_window_dims = [0, 2, 3], inserted_window_dims = [1], scatter_dims_to_operand_dims = [1], index_vector_dim = 1>, unique_indices = false} : (tensor<1x500x4x4xf32>, tensor<100x1xi64>, tensor<1x100x4x4xf32>) -> tensor<1x500x4x4xf32> %7 = "stablehlo.scatter"(%arg1, %5, %arg4) ({
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->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) -
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) -
tensorflow/compiler/mlir/lite/stablehlo/tests/legalize_hlo.mlir
// CHECK-SAME: indices_are_sorted = false, scatter_dimension_numbers = #mhlo.scatter<update_window_dims = [1], inserted_window_dims = [0], scatter_dims_to_operand_dims = [0], index_vector_dim = 1>, unique_indices = false // CHECK-SAME: }> ({ // CHECK: ^bb0(%[[VAL_4:.*]]: tensor<f32>, %[[VAL_5:.*]]: tensor<f32>):
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/flatbuffer_export.cc
auto inserted_window_dims = builder_.CreateVector(inserted_window_dims_vec); auto scatter_dims_to_operand_dims = builder_.CreateVector(scatter_dims_to_operand_dims_vec); auto options = tflite::CreateStablehloScatterOptions( builder_, indices_are_sorted, update_window_dims, inserted_window_dims, scatter_dims_to_operand_dims, index_vector_dim, unique_indices, subgraph_index);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (0) -
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) -
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) -
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf.mlir
// CHECK-SAME: indices_are_sorted = false, // CHECK-SAME: scatter_dimension_numbers = // CHECK-SAME: update_window_dims = [2] // CHECK-SAME: inserted_window_dims = [0] // CHECK-SAME: scatter_dims_to_operand_dims = [0] // CHECK-SAME: index_vector_dim = 2 // CHECK-SAME: unique_indices = false // CHECK: ^{{.*}}([[LHS:%.*]]: tensor<f32>, [[RHS:%.*]]: tensor<f32>):
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/schema/schema_generated.h
{ auto _e = scatter_dims_to_operand_dims(); if (_e) { _o->scatter_dims_to_operand_dims.resize(_e->size()); for (::flatbuffers::uoffset_t _i = 0; _i < _e->size(); _i++) { _o->scatter_dims_to_operand_dims[_i] = _e->Get(_i); } } else { _o->scatter_dims_to_operand_dims.resize(0); } } { auto _e = index_vector_dim(); _o->index_vector_dim = _e; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 1M bytes - Viewed (0)