Searched refs:weights_inputs (Results 1 – 1 of 1) sorted by relevance
263 OpMutableInputList weights_inputs; in Compute() local265 context->mutable_input_list("weights", &weights_inputs)); in Compute()269 auto prox_w = weights_inputs.at(i, /*lock_held=*/true).flat<float>(); in Compute()275 if (weights_inputs.size() > 0) { in Compute()277 for (int i = 0; i < weights_inputs.size(); ++i) { in Compute()278 num_weights += weights_inputs.at(i, /*lock_held=*/true).NumElements(); in Compute()281 const int64 kCostPerUnit = (num_weights * 50) / weights_inputs.size(); in Compute()285 weights_inputs.size(), kCostPerUnit, do_work); in Compute()