- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for TopKV2 (0.27 sec)
-
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
// // values, indices = tfl.topkv2(%inputs, K) // %1 = tfl.slice(values, 0, k) // %2 = tfl.slice(indices,0, k) // .... (values and indices only used for %1 and %2) // // %1 or %2 can be absent. If values and indices are only used here, // this pattern can be replaced with (conceptually) // // %values, %indices = tfl.topkv2(%inputs, k) // replace all use of %1 with values
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/tf2xla/tests/legalize-tf.mlir
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/stablehlo/tests/legalize_hlo.mlir
// CHECK: %[[K:.*]] = "tf.Const"() <{value = dense<6> : tensor<i32>}> : () -> tensor<i32> // CHECK: %[[VALUES:.*]], %[[INDICES:.*]] = "tf.TopKV2"(%[[ARG_0]], %[[K]]) <{sorted = true}> : (tensor<3x6xf32>, tensor<i32>) -> (tensor<3x6xf32>, tensor<3x6xi32>) // CHECK: return %[[VALUES]], %[[INDICES]] : tensor<3x6xf32>, tensor<3x6xi32> // CHECK: }
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/jit/mark_for_compilation_pass.cc
"_FusedBatchNormEx", "FusedBatchNormGrad", "FusedBatchNormGradV2", "FusedBatchNormGradV3"}}, {"Conv", {"_FusedConv2D"}}, {"SORT", {"TopKV2"}}, // XLA version much faster then TF version. {"MISC", // clang-format off {"ApproxTopK", "BroadcastTo", "ExpandDims", "Fill", "NoOp",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 12:19:41 UTC 2024 - 85.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
/*outfeed_config=*/rewriter.getStringAttr("")); rewriter.eraseOp(op); return success(); } }; // Converts tf.TopKV2 to chlo.top_k. class ConvertTopKV2Op : public OpRewritePattern<TF::TopKV2Op> { public: using OpRewritePattern::OpRewritePattern; LogicalResult matchAndRewrite(TF::TopKV2Op op,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td
TF_Float32Tensor:$input, I64Attr:$k ); let results = (outs TF_Float32Tensor:$topk, TF_Int32Tensor:$topk_indices ); } def TF_TopKV2Op : TF_Op<"TopKV2", [Pure]> { let summary = [{ Finds values and indices of the `k` largest elements for the last dimension. }]; let description = [{
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K bytes - Viewed (0)