Searched refs:UseBufferForWeights (Results 1 – 2 of 2) sorted by relevance
31 bool UseBufferForWeights(const GpuInfo& gpu_info) { in UseBufferForWeights() function79 const bool weights_are_buffer = UseBufferForWeights(gpu_info); in GetFCFCAddKernelCode()178 UseBufferForWeights(gpu_info)); in CreateFCFCAdd()180 UseBufferForWeights(gpu_info)); in CreateFCFCAdd()
34 bool UseBufferForWeights(const GpuInfo& gpu_info) { in UseBufferForWeights() function86 const bool weights_are_buffer = UseBufferForWeights(gpu_info); in GetFullyConnectedKernelCode()161 result.UploadWeights(attr.weights, UseBufferForWeights(gpu_info)); in CreateFullyConnected()