- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for top_k (0.11 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
if (!backend_config) { return op.emitOpError() << "Missing backend_config attribute"; } for (const auto& attr : backend_config) { auto name = attr.getName(); if (!(name == "top_k" || name == "reduction_dim" || name == "recall_target" || name == "aggregate_to_topk" || name == "reduction_input_size_override" || name == "is_fallback")) { return op.emitOpError()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/quantize_model_test.py
) def multiple_output_ops( self, input_tensor: core.Tensor ) -> Mapping[str, core.Tensor]: k = array_ops.constant(4, dtype=dtypes.int32) values, indices = nn_ops.top_k(input_tensor, k, name='TopK') adj_values = values + 2 return {'indices': indices, 'adj_values': adj_values, 'values': values} @def_function.function( input_signature=[
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 03:36:50 UTC 2024 - 235.6K 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/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)