Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for GetVariableToShapeMap (0.12 sec)

  1. tensorflow/c/checkpoint_reader.cc

        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(
            reader_->GetVariableToDataTypeMap()));
      }
    }
    
    bool CheckpointReader::HasTensor(const string& name) const {
    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

      const string DebugString() const;
    
      // Returns a map from variable names to their shapes.  Slices of a partitioned
      // tensor are combined into a single entry.
      const TensorSliceReader::VarToShapeMap& GetVariableToShapeMap() const;
    
      // 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;
    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