Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for CheckpointReader (0.3 sec)

  1. tensorflow/c/checkpoint_reader.h

      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)
  2. tensorflow/c/checkpoint_reader.cc

    const TensorSliceReader::VarToDataTypeMap&
    CheckpointReader::GetVariableToDataTypeMap() const {
      CHECK(var_to_data_type_map_);
      return *var_to_data_type_map_;
    }
    
    const string CheckpointReader::DebugString() const {
      if (reader_ != nullptr) return reader_->DebugString();
      return v2_reader_->DebugString();
    }
    
    void CheckpointReader::GetTensor(
        const string& name, std::unique_ptr<tensorflow::Tensor>* out_tensor,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 5.6K bytes
    - Viewed (0)
  3. tensorflow/c/c_api_experimental.cc

      status->status = tensorflow::errors::Internal(errMsg);
    }
    
    struct TF_CheckpointReader : public tensorflow::checkpoint::CheckpointReader {
      using tensorflow::checkpoint::CheckpointReader::CheckpointReader;
      std::vector<std::string> variable_list;
    };
    
    TF_CheckpointReader* TF_NewCheckpointReader(const char* filename,
                                                TF_Status* status) {
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 29.5K bytes
    - Viewed (0)
  4. tensorflow/c/c_api_experimental.h

    TF_CAPI_EXPORT extern void TF_MakeInternalErrorStatus(TF_Status* status,
                                                          const char* errMsg);
    
    // TF_NewCheckpointReader() return the CheckpointReader that can be use to
    // investigate or load the variable from the checkpoint file
    typedef struct TF_CheckpointReader TF_CheckpointReader;
    TF_CAPI_EXPORT extern TF_CheckpointReader* TF_NewCheckpointReader(
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Apr 27 21:07:00 UTC 2023
    - 15.1K bytes
    - Viewed (0)
Back to top