Searched refs:tensor_data_ (Results 1 – 4 of 4) sorted by relevance
315 tensor_data_[id] = t; in AddConstSparseInput()420 tensor_data_[id] = t;519 const std::vector<int>& GetShape(int id) { return tensor_data_.at(id).shape; } in GetShape()521 float GetScale(int id) { return tensor_data_.at(id).scale; } in GetScale()522 int32_t GetZeroPoint(int id) { return tensor_data_.at(id).zero_point; } in GetZeroPoint()746 tensor_data_[id] = t;860 tensor_data_[id] = t; in AddTensorPerChannelQuant()917 std::map<int, TensorData> tensor_data_; variable
162 std::vector<at::Tensor> tensor_data_; member in torch::jit::__anoncbf17dd60111::FlatbufferSerializer414 for (auto td : tensor_data_) { in serializeModule()445 static_cast<int32_t>(tensor_data_.size()), in serializeModule()654 storage_index = tensor_data_.size(); in tensorToFB()656 tensor_data_.push_back(tensor); in tensorToFB()
156 return tensor_data_; in tensorData()271 std::vector<at::Tensor> tensor_data_; variable
310 : std::to_string(tensor_data_.size()); in pushStorageOfTensor()323 tensor_data_.push_back(tensor); in pushStorageOfTensor()