- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for TF_CheckpointReaderGetTensor (0.17 sec)
-
tensorflow/c/c_api_experimental.h
TF_CAPI_EXPORT extern int TF_CheckpointReaderGetVariableNumDims( TF_CheckpointReader* reader, const char* name); // Load the weight of a variable TF_CAPI_EXPORT extern TF_Tensor* TF_CheckpointReaderGetTensor( TF_CheckpointReader* reader, const char* name, TF_Status* status); // TF_NewAttrBuilder() returns an object that you can set attributes on as
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 27 21:07:00 UTC 2023 - 15.1K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.cc
const char* name) { const auto& m = reader->GetVariableToDataTypeMap(); return static_cast<TF_DataType>(m.at(name)); } TF_Tensor* TF_CheckpointReaderGetTensor(TF_CheckpointReader* reader, const char* name, TF_Status* status) { std::unique_ptr<tensorflow::Tensor> tensor; reader->GetTensor(name, &tensor, status);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 29.4K bytes - Viewed (0)