Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for CPUDevice (0.1 sec)

  2. github.com/tensorflow/tensorflow

    tensorflow/compiler/mlir/quantization/tensorflow/calibrator/custom_aggregator_op.cc 

        OP_REQUIRES_OK(context, context->allocate_output("max", {}, &max_output));
    min_output->scalar<float>().device(
        context->template eigen_device<CPUDevice>()) = input_flat.minimum();
    max_output->scalar<float>().device(
        context->template eigen_device<CPUDevice>()) = input_flat.maximum();

    // Calculate histogram statistics.
    const int32_t num_bins = calib_opts_.calibration_parameters().num_bins();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 01:09:50 UTC 2024
    - 6.2K bytes
    - Viewed (0)
Back to top