Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for ReadBinaryProto (0.13 sec)

  1. tensorflow/compiler/mlir/quantization/tensorflow/calibrator/calibration_statistics_saver_op_test.cc

      AddInputFromArray<int64_t>(TensorShape({0}), {});
    
      TF_CHECK_OK(RunOpKernel());
      kernel_.reset();
    
      CalibrationStatisticsMap statistics_map;
      TF_CHECK_OK(
          ReadBinaryProto(Env::Default(), output_file_path, &statistics_map));
      ASSERT_THAT(statistics_map.statistics(), SizeIs(1));
      ASSERT_THAT(statistics_map.statistics(), ElementsAre(Key("1")));
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 13 01:31:23 UTC 2024
    - 11.4K bytes
    - Viewed (0)
  2. tensorflow/cc/saved_model/fingerprinting.cc

        absl::string_view export_dir, std::string pb_file) {
      // Version of the code that produced the fingerprint.
      const int kFingerprintProducer = 1;
    
      SavedModel saved_model;
      TF_RETURN_IF_ERROR(ReadBinaryProto(Env::Default(), pb_file, &saved_model));
    
      // Create a copy of `metagraph` which will be used and mutated for fingerprint
      // computation.
      FingerprintDef fingerprint_def;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 26 00:24:44 UTC 2024
    - 9.8K bytes
    - Viewed (0)
  3. tensorflow/compiler/aot/compile.cc

    }
    
    static Status ReadProtoFile(const string& fname, protobuf::Message* proto) {
      if (absl::EndsWith(fname, ".pbtxt")) {
        return ReadTextProto(Env::Default(), fname, proto);
      } else {
        return ReadBinaryProto(Env::Default(), fname, proto);
      }
    }
    
    static absl::once_flag targets_init;
    
    static void InitializeTargets() {
      // Initialize all LLVM targets so we can cross compile.
    #if TF_LLVM_AARCH32_AVAILABLE
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 15 08:28:57 UTC 2024
    - 11.9K bytes
    - Viewed (0)
Back to top