Searched refs:cost_per_coeff (Results 1 – 1 of 1) sorted by relevance
175 double output_size, const TensorOpCost& cost_per_coeff, int max_threads) { in numThreads() argument176 double cost = totalCost(output_size, cost_per_coeff); in numThreads()185 double output_size, const TensorOpCost& cost_per_coeff) { in taskSize() argument186 return totalCost(output_size, cost_per_coeff) / kTaskSize; in taskSize()191 double output_size, const TensorOpCost& cost_per_coeff) { in totalCost() argument205 cost_per_coeff.total_cost(kLoadCycles, kStoreCycles, in totalCost()