- Sort Score
- Num 10 results
- Language All
Results 1 - 2 of 2 for GetVariableToDataTypeMap (0.13 seconds)
-
tensorflow/c/checkpoint_reader.cc
} var_to_shape_map_.reset( new TensorSliceReader::VarToShapeMap(reader_->GetVariableToShapeMap())); var_to_data_type_map_.reset(new TensorSliceReader::VarToDataTypeMap( reader_->GetVariableToDataTypeMap())); } } bool CheckpointReader::HasTensor(const string& name) const { if (reader_ != nullptr) { return reader_->HasTensor(name, nullptr, nullptr); } return v2_reader_->Contains(name); }Created: Tue Apr 07 12:39:13 GMT 2026 - Last Modified: Tue Feb 17 18:49:14 GMT 2026 - 5.4K bytes - Click Count (0) -
tensorflow/c/checkpoint_reader.h
// Returns a map from variable names to their data types. Slices of a // partitioned tensor are combined into a single entry. const TensorSliceReader::VarToDataTypeMap& GetVariableToDataTypeMap() const; // Attempts to look up the tensor named "name" and stores the found result in // "out_tensor". void GetTensor(const std::string& name, std::unique_ptr<tensorflow::Tensor>* out_tensor,
Created: Tue Apr 07 12:39:13 GMT 2026 - Last Modified: Wed Jan 07 04:56:09 GMT 2026 - 3.1K bytes - Click Count (0)