Searched refs:serialized_model_tensor (Results 1 – 2 of 2) sorted by relevance
37 at::Tensor serialized_model_tensor, in init() argument41 std::move(serialized_model_tensor), in init()48 at::Tensor serialized_model_tensor, in init2() argument67 TORCH_CHECK(serialized_model_tensor.is_contiguous()); in init2()71 serialized_model_tensor.scalar_type() == at::ScalarType::Byte in init2()72 ? serialized_model_tensor.data_ptr<uint8_t>() in init2()73 : reinterpret_cast<uint8_t*>(serialized_model_tensor.data_ptr<int32_t>()); in init2()76 serialized_model_tensor.nbytes() in init2()
41 at::Tensor serialized_model_tensor,46 at::Tensor serialized_model_tensor,