Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for top_k (0.08 sec)

  1. 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)
  2. tensorflow/compiler/mlir/lite/stablehlo/tests/legalize_hlo.mlir

        backend_config = {
          aggregate_to_topk = true,
          is_fallback = true,
          recall_target = 8.500000e-01 : f32,
          reduction_dim = 1 : i64,
          reduction_input_size_override = -1 : i64,
          top_k = 4 : i64}
        } : (tensor<1x4xf32>, tensor<1x4xi32>, tensor<f32>, tensor<i32>) -> (tensor<1x4xf32>, tensor<1x4xi32>)
      func.return %0#0, %0#1 : tensor<1x4xf32>, tensor<1x4xi32>
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 29 07:26:59 UTC 2024
    - 340.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tf2xla/tests/legalize-tf.mlir

    func.func @topk_v2(%input: tensor<16x16xf32>) -> (tensor<16x8xf32>, tensor<16x8xi32>) {
      %k = "tf.Const"() {value = dense<8> : tensor<i32>} : () -> tensor<i32>
    
      // CHECK:     chlo.top_k(%[[INPUT]], k = 8)
      %0:2 = "tf.TopKV2"(%input, %k): (tensor<16x16xf32>, tensor<i32>) -> (tensor<16x8xf32>, tensor<16x8xi32>)
      func.return %0#0, %0#1: tensor<16x8xf32>, tensor<16x8xi32>
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 06 18:46:23 UTC 2024
    - 335.5K bytes
    - Viewed (0)
  4. docs/en/docs/release-notes.md

    * Add support for setting a `default_response_class` in the `FastAPI` instance or in `include_router`. Initial PR [#467](https://github.com/tiangolo/fastapi/pull/467) by [@toppk](https://github.com/toppk).
    * Add support for type annotations using strings and `from __future__ import annotations`. PR [#451](https://github.com/tiangolo/fastapi/pull/451) by [@dmontagu](https://github.com/dmontagu).
    
    ## 0.38.1
    
    Registered: Mon Jun 17 08:32:26 UTC 2024
    - Last Modified: Fri Jun 14 15:07:37 UTC 2024
    - 395.4K bytes
    - Viewed (0)
Back to top