Searched refs:weights_inputs (Results 1 - 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Dsdca_ops.cc251 OpMutableInputList weights_inputs; variable
253 context->mutable_input_list("weights", &weights_inputs));
257 auto prox_w = weights_inputs.at(i, /*lock_held=*/true).flat<float>();
263 if (weights_inputs.size() > 0) {
265 for (int i = 0; i < weights_inputs.size(); ++i) {
266 num_weights += weights_inputs.at(i, /*lock_held=*/true).NumElements();
269 const int64 kCostPerUnit = (num_weights * 50) / weights_inputs.size();
273 weights_inputs.size(), kCostPerUnit, do_work);

Completed in 97 milliseconds