Searched refs:local_mem_size (Results 1 – 3 of 3) sorted by relevance
740 cl_ulong local_mem_size; in skc_device_create_kernels() local745 sizeof(local_mem_size), in skc_device_create_kernels()746 &local_mem_size, in skc_device_create_kernels()750 (unsigned long)local_mem_size); in skc_device_create_kernels()
445 const int local_mem_size = in GenerateConv() local612 std::to_string(local_mem_size) + "];\n"; in GenerateConv()908 /*global_offset_name*/ "", local_mem_size); in GenerateConv()914 total_work_items, local_mem_size); in GenerateConv()916 int parts = local_mem_size / simd_size; in GenerateConv()917 int reminder = local_mem_size % simd_size; in GenerateConv()964 c += " filters_loc += " + std::to_string(local_mem_size) + ";\n"; in GenerateConv()
181 const int local_mem_size = in GenerateConvolution() local305 c += " threadgroup FLT4 weights_cache[" + std::to_string(local_mem_size) + in GenerateConvolution()394 total_work_items, local_mem_size); in GenerateConvolution()397 int parts = local_mem_size / simd_size; in GenerateConvolution()398 int reminder = local_mem_size % simd_size; in GenerateConvolution()