- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 21 for top_k (0.29 sec)
-
guava/src/com/google/common/collect/Comparators.java
checkNotNull(comparator); return Collector.of( () -> TopKSelector.<T>least(k, comparator), TopKSelector::offer, TopKSelector::combine, TopKSelector::topK, Collector.Characteristics.UNORDERED); } /** * Returns a {@code Collector} that returns the {@code k} greatest (relative to the specified
Registered: Wed Jun 12 16:38:11 UTC 2024 - Last Modified: Thu Apr 11 08:42:51 UTC 2024 - 10.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tf_tfl_passes.cc
// to be consistent with other entrypoints. pass_manager.addPass(mlir::mhlo::createHloLegalizeToStablehloPass()); // Decompose CHLO into StableHLO ops // TODO(b/331843141): There are some CHLO's like TopK which we could instead // lower to TFL ops. mlir::stablehlo::experimental::createChloLegalizeToStablehloPipeline( pass_manager); pass_manager.addPass(mlir::odml::CreateTransposeCommuteOpsPass());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 18:45:51 UTC 2024 - 25.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_patterns.td
(TFL_MeanOp $arg0, $arg1, $arg2)>; def LegalizeSum : Pat<(TF_SumOp $arg, $axes, BoolAttr:$arg2), (TFL_SumOp $arg, (CreateTFCastToInt32Op $axes), $arg2)>; // TopK in TFL is always sorted so we ignore that attribute here. def LegalizeTopKV2 : Pat<(TF_TopKV2Op $input, $k, $ignored_sorted), (TFL_TopKV2Op $input, $k)>;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 04 13:30:42 UTC 2024 - 28.5K bytes - Viewed (0) -
android/guava/src/com/google/common/collect/Ordering.java
} list.trimToSize(); return Collections.unmodifiableList(list); } else { TopKSelector<E> selector = TopKSelector.least(k, this); selector.offerAll(iterator); return selector.topK(); } } /** * Returns the {@code k} greatest elements of the given iterable according to this ordering, in * order from greatest to least. If there are fewer than {@code k} elements present, all will be
Registered: Wed Jun 12 16:38:11 UTC 2024 - Last Modified: Tue May 28 18:11:09 UTC 2024 - 39.4K bytes - Viewed (0) -
guava/src/com/google/common/collect/Ordering.java
} list.trimToSize(); return Collections.unmodifiableList(list); } else { TopKSelector<E> selector = TopKSelector.least(k, this); selector.offerAll(iterator); return selector.topK(); } } /** * Returns the {@code k} greatest elements of the given iterable according to this ordering, in * order from greatest to least. If there are fewer than {@code k} elements present, all will be
Registered: Wed Jun 12 16:38:11 UTC 2024 - Last Modified: Tue May 28 18:11:09 UTC 2024 - 39.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/ops.mlir
} // ----- // CHECK-LABEL: topk func.func @topk(%arg0: tensor<8xf32>, %arg1: tensor<i32>) -> (tensor<?xf32>, tensor<?xi32>) { %0, %1 = "tfl.topk_v2"(%arg0, %arg1) : (tensor<8xf32>, tensor<i32>) -> (tensor<?xf32>, tensor<?xi32>) func.return %0, %1: tensor<?xf32>, tensor<?xi32> } // ----- // CHECK-LABEL: topk
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 19:09:08 UTC 2024 - 189.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/legalize-tf.mlir
// CHECK-LABEL:sin // CHECK: "tfl.sin"(%arg0) : (tensor<f32>) -> tensor<f32> } func.func @topk(%arg0: tensor<8xf32>, %arg1: tensor<i32>) -> (tensor<?xf32>, tensor<?xi32>) { %0, %1 = "tf.TopKV2"(%arg0, %arg1) : (tensor<8xf32>, tensor<i32>) -> (tensor<?xf32>, tensor<?xi32>) func.return %0, %1: tensor<?xf32>, tensor<?xi32> // CHECK-LABEL: topk // CHECK: "tfl.topk_v2"(%arg0, %arg1) // CHECK: return }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 05 01:54:33 UTC 2024 - 153.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 102.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td
This outputs a `batch_size` bool array, an entry `out[i]` is `true` if the prediction for the target class is among the top `k` predictions among all predictions for example `i`. Note that the behavior of `InTopK` differs from the `TopK` op in its handling of ties; if multiple classes have the same prediction value and straddle the top-`k` boundary, all of those classes are considered to be in the top `k`. More formally, let
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.td
TFL_OperandHasRankAtLeast<0, 1>, TFL_OperandHasRank<1, 0>, PredOpTrait<"result and input element type match", TFL_TCresVTEtIsSameAsOp<0,0>>, SameOperandsAndResultsScale]> { let summary = "TopK operator"; let description = [{ Returns the top `k` largest element along each last dimensional slice of `input` and the indices of values within the last dimension of the input tensor.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 19:09:08 UTC 2024 - 186K bytes - Viewed (0)