Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for var_to_data_type_map_ (0.23 sec)

  1. tensorflow/c/checkpoint_reader.cc

        var_to_shape_map_.swap(result.first);
        var_to_data_type_map_.swap(result.second);
      } else {
        reader_.reset(new TensorSliceReader(filename));
        if (!reader_->status().ok()) {
          tsl::Set_TF_Status_from_Status(status, reader_->status());
          return;
        }
        var_to_shape_map_.reset(
            new TensorSliceReader::VarToShapeMap(reader_->GetVariableToShapeMap()));
        var_to_data_type_map_.reset(new TensorSliceReader::VarToDataTypeMap(
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 5.6K bytes
    - Viewed (0)
  2. tensorflow/c/checkpoint_reader.h

      std::unique_ptr<TensorSliceReader> reader_;
      std::unique_ptr<BundleReader> v2_reader_;
    
      std::unique_ptr<TensorSliceReader::VarToShapeMap> var_to_shape_map_;
      std::unique_ptr<TensorSliceReader::VarToDataTypeMap> var_to_data_type_map_;
    
      CheckpointReader(const CheckpointReader&) = delete;
      void operator=(const CheckpointReader&) = delete;
    };
    
    }  // namespace checkpoint
    }  // namespace tensorflow
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Oct 12 08:49:52 UTC 2023
    - 3.1K bytes
    - Viewed (0)
Back to top