Searched refs:gpu_storage (Results 1 – 2 of 2) sorted by relevance
188 LinearStorage gpu_storage; in CreateCLObject() local190 gpu_storage.CreateFromTensorLinearDescriptor(*linear_desc, context)); in CreateCLObject()191 *result = absl::make_unique<LinearStorage>(std::move(gpu_storage)); in CreateCLObject()
145 LinearStorage gpu_storage; in CreateMetalObject() local147 gpu_storage.CreateFromTensorLinearDescriptor(*linear_desc, device)); in CreateMetalObject()148 *result = absl::make_unique<LinearStorage>(std::move(gpu_storage)); in CreateMetalObject()