Search Options

Display Count
Sort
Preferred Language
Advanced Search

Results 1 - 2 of 2 for GetVariableToDataTypeMap (0.17 seconds)

  1. 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)
  2. 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)
Back to Top