Searched defs:gpu_data (Results 1 – 8 of 8) sorted by relevance
39 std::vector<float> gpu_data; in TEST_F() local52 std::vector<half> gpu_data; in TEST_F() local
39 std::vector<float> gpu_data; in TEST_F() local51 std::vector<half> gpu_data; in TEST_F() local
104 float* gpu_data = reinterpret_cast<float*>(data.data()); in UploadLinearData() local114 half* gpu_data = reinterpret_cast<half*>(data.data()); in UploadLinearData() local
911 float* gpu_data = reinterpret_cast<float*>(data.data()); in UploadData() local914 half* gpu_data = reinterpret_cast<half*>(data.data()); in UploadData() local
85 float4* gpu_data = reinterpret_cast<float4*>(desc.data.data()); in UploadData() local93 half4* gpu_data = reinterpret_cast<half4*>(desc.data.data()); in UploadData() local
240 float* gpu_data = reinterpret_cast<float*>(desc.data.data()); in UploadBias() local245 half* gpu_data = reinterpret_cast<half*>(desc.data.data()); in UploadBias() local
607 float* gpu_data = reinterpret_cast<float*>(result.data()); in ReorderBiasesForConv() local612 half* gpu_data = reinterpret_cast<half*>(result.data()); in ReorderBiasesForConv() local
36 std::vector<float> gpu_data; in UploadWeights() local