Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for HasTensor (0.07 sec)

  1. tensorflow/c/checkpoint_reader.cc

        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);
    }
    
    const TensorSliceReader::VarToShapeMap&
    CheckpointReader::GetVariableToShapeMap() 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/c_api_experimental.cc

    }
    
    void TF_DeleteCheckpointReader(TF_CheckpointReader* reader) { delete reader; }
    
    int TF_CheckpointReaderHasTensor(TF_CheckpointReader* reader,
                                     const char* name) {
      return reader->HasTensor(name);
    }
    
    const char* TF_CheckpointReaderGetVariable(TF_CheckpointReader* reader,
                                               int index) {
      return reader->variable_list[index].c_str();
    }
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 29.5K bytes
    - Viewed (0)
Back to top