Searched refs:sparse_weights_inputs (Results 1 – 1 of 1) sorted by relevance
99 OpInputList sparse_weights_inputs; in Initialize() local101 context->input_list("sparse_weights", &sparse_weights_inputs)); in Initialize()114 for (int i = 0; i < sparse_weights_inputs.size(); ++i) { in Initialize()117 i, sparse_weights_inputs[i].shape(), &delta_t)); in Initialize()123 sparse_weights_inputs[i].shaped<float, 2>( in Initialize()124 {1, sparse_weights_inputs[i].NumElements()}), in Initialize()