Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 9 of 9 for global_ml_ (0.35 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/lower_globals_to_ml_program.cc

      for (auto globalTensor : module.getOps<tf_saved_model::GlobalTensorOp>()) {
        auto exportedNames = tf_saved_model::GetExportedNames(globalTensor);
        std::string name;
        if (exportedNames.empty()) {
          name = "global_ml_" + globalTensor.getSymName().str();
        } else if (exportedNames.size() == 1) {
          name = exportedNames[0].str();
        } else {
          return globalTensor.emitError()
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.6K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/tensorflow/calibrator/calibration_statistics_collector_min_max.cc

    namespace calibrator {
    
    void CalibrationStatisticsCollectorMinMax::ClearData() {
      // global_min will be updated by std::min(global_min, input_value) so
      // it is initialized with the value numeric_limits<float>::max().
      min_max_statistics_.set_global_min(std::numeric_limits<float>::max());
    
      // global_max will be updated by std::max(global_max, input_value) so it
      // is initialized with the value numeric_limits<float>::lowest().
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Apr 05 03:57:26 UTC 2024
    - 2.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/quantization/tensorflow/calibrator/calibration_statistics.proto

    // Calibration algorithm's collecting statistics.
    // NEXT_ID: 4
    message CalibrationStatistics {
      message MinMaxStatistics {
        // global minimum of all sample datasets.
        float global_min = 1;
    
        // global maximum of all sample datasets.
        float global_max = 2;
      }
    
      message AverageMinMaxStatistics {
        // sum of batch's minimum in each sample dataset.
        float min_sum = 1;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 16 04:33:52 UTC 2024
    - 1.3K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/quantization/tensorflow/calibrator/calibration_statistics_collector_test.cc

      std::optional<CalibrationStatistics> statistics = collector.GetStatistics();
    
      EXPECT_TRUE(statistics.has_value());
      EXPECT_EQ(statistics.value().min_max_statistics().global_min(), -5.0f);
      EXPECT_EQ(statistics.value().min_max_statistics().global_max(), 10.0f);
    }
    
    TEST(CalibrationStatisticsCollectorTest, SimpleAverageMinMax) {
      auto collector = CalibrationStatisticsCollectorAverageMinMax();
    
      collector.Collect(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 16 04:33:52 UTC 2024
    - 12.6K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/quantization/tensorflow/calibrator/calibration_algorithm_test.py

            calibration_method=_CalibrationMethod.CALIBRATION_METHOD_MIN_MAX
        )
        statistics = calib_stats_pb2.CalibrationStatistics()
        statistics.min_max_statistics.global_min = 1.0
        statistics.min_max_statistics.global_max = 5.0
    
        min_value, max_value = calibration_algorithm.get_min_max_value(
            statistics, calib_opts
        )
        self.assertAllEqual((min_value, max_value), (1.0, 5.0))
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Mar 11 19:29:56 UTC 2024
    - 5K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/quantization/tensorflow/calibrator/calibration_statistics_saver_op_test.cc

      const CalibrationStatistics& stats = statistics_map.statistics().at("1");
      ASSERT_TRUE(stats.has_min_max_statistics());
      EXPECT_FLOAT_EQ(stats.min_max_statistics().global_min(), 1.f);
      EXPECT_FLOAT_EQ(stats.min_max_statistics().global_max(), 5.f);
    }
    
    TEST_F(CalibrationStatisticsSaverTest, SimpleAverageMinMax) {
      std::vector<std::string> ids{"1"};
      std::vector<int32_t> calibration_methods{
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 13 01:31:23 UTC 2024
    - 11.4K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/quantization/tensorflow/calibrator/calibration_algorithm.py

        Returns:
          (min_value, max_value): Min and max calculated using MinMax
        """
        return (
            self._statistics.min_max_statistics.global_min,
            self._statistics.min_max_statistics.global_max,
        )
    
    
    @_implements(_CalibrationMethod.CALIBRATION_METHOD_AVERAGE_MIN_MAX)
    class _AverageMinMax(_CalibrationAlgorithmBase):
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Mar 11 19:29:56 UTC 2024
    - 14.7K bytes
    - Viewed (0)
  8. tensorflow/compiler/jit/xla_device.h

      // compute streams used in each XLA device. It is only used if
      // `use_global_compute_stream` in `XlaDevice::Options` is set to true.
      static mutex global_mu_;
      static std::vector<std::shared_ptr<se::Stream>>* global_compute_streams_
          TF_GUARDED_BY(global_mu_);
    };
    
    // Builds OpKernel registrations on 'device' for the JIT operators
    // registered on 'jit_device'. Returns ownership of a XlaDeviceOpRegistrations
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 09:53:30 UTC 2024
    - 13.4K bytes
    - Viewed (0)
  9. tensorflow/compiler/jit/xla_device.cc

                                               const Metadata** metadata) {
      return GetMetadataFromDevice(ctx->device(), metadata);
    }
    
    /* static */ mutex XlaDevice::global_mu_(LINKER_INITIALIZED);
    /* static */ std::vector<std::shared_ptr<se::Stream>>*
        XlaDevice::global_compute_streams_ =
            new std::vector<std::shared_ptr<se::Stream>>;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 20 21:05:42 UTC 2024
    - 24.3K bytes
    - Viewed (0)
Back to top