- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for scatter_dims_to_operand_dims (0.45 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/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/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)