Searched defs:gpu_data (Results 1 – 9 of 9) sorted by relevance
344 half* gpu_data = reinterpret_cast<half*>(data_.data()); in UploadData() local347 T* gpu_data = reinterpret_cast<T*>(data_.data()); in UploadData() local352 half* gpu_data = reinterpret_cast<half*>(data_.data()); in UploadData() local355 T* gpu_data = reinterpret_cast<T*>(data_.data()); in UploadData() local365 half* gpu_data = reinterpret_cast<half*>(data_.data()); in DownloadData() local368 T* gpu_data = reinterpret_cast<T*>(data_.data()); in DownloadData() local
384 float* gpu_data = reinterpret_cast<float*>(result.data()); in ReorderWeightsForConv() local389 half* gpu_data = reinterpret_cast<half*>(result.data()); in ReorderWeightsForConv() local402 float* gpu_data = reinterpret_cast<float*>(result.data()); in ReorderBiasesForConv() local407 half* gpu_data = reinterpret_cast<half*>(result.data()); in ReorderBiasesForConv() local
88 float4* gpu_data = reinterpret_cast<float4*>(desc.data.data()); in UploadData() local96 half4* gpu_data = reinterpret_cast<half4*>(desc.data.data()); in UploadData() local
248 float* gpu_data = reinterpret_cast<float*>(desc.data.data()); in UploadBias() local253 half* gpu_data = reinterpret_cast<half*>(desc.data.data()); in UploadBias() local
39 std::vector<float> gpu_data; in TEST_F() local51 std::vector<half> gpu_data; in TEST_F() local
80 DataType* gpu_data = static_cast<DataType*>(sycl_device.allocate(in1.size()*sizeof(DataType))); in test_sycl_mem_sync() local106 DataType* gpu_data = static_cast<DataType*>(sycl_device.allocate(full_size * sizeof(DataType))); in test_sycl_mem_sync_offsets() local165 DataType* gpu_data = static_cast<DataType*>(sycl_device.allocate(full_size * sizeof(DataType))); in test_sycl_memset_offsets() local
64 DataType *gpu_data = static_cast<DataType *>( in test_unary_builtins_for_scalar() local171 DataType *gpu_data = static_cast<DataType *>( in test_unary_builtins_return_bool() local
49 DataType* gpu_data = static_cast<DataType*>(sycl_device.allocate(sizeDim1*sizeof(DataType))); in test_device_exceptions() local
348 def get_ext_support(ext_name, gpu_data): argument364 def format_md_gpu_info(gpu_data): argument