Searched refs:layer_weights (Results 1 – 5 of 5) sorted by relevance
| /external/libaom/libaom/av1/encoder/ |
| D | ml.c | 31 const float *layer_weights = nn_config->weights[layer]; in av1_nn_predict_c() local
39 val += layer_weights[node * num_input_nodes + i] * input_nodes[i]; in av1_nn_predict_c()
50 const float *layer_weights = nn_config->weights[num_layers]; in av1_nn_predict_c() local
55 val += layer_weights[node * num_input_nodes + i] * input_nodes[i]; in av1_nn_predict_c()
|
| /external/libaom/libaom/av1/encoder/x86/ |
| D | ml_sse3.c | 162 const float *layer_weights = nn_config->weights[layer]; in av1_nn_predict_sse3() local
175 &layer_weights[out * num_inputs + in], &out_h, in av1_nn_predict_sse3()
187 &layer_weights[out * num_inputs + in], &outputs, in av1_nn_predict_sse3()
198 &layer_weights[out * num_inputs + in], &outputs, in av1_nn_predict_sse3()
209 &layer_weights[out * num_inputs + in], &total); in av1_nn_predict_sse3()
219 &layer_weights[out * num_inputs + in], &total); in av1_nn_predict_sse3()
232 _mm_load1_ps(&layer_weights[num_inputs * out + in_node]); in av1_nn_predict_sse3()
|
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/ops/ |
| D | cudnn_rnn_ops.py | 297 layer_weights = cu_weights[i * layer_weights_num:(i + 1) *
302 layer_weights, layer_biases, tf_weights, tf_biases)
304 fw_weights = layer_weights[:len(layer_weights) // 2]
305 bw_weights = layer_weights[len(layer_weights) // 2:]
369 layer_weights = weights[i * layer_weights_num:(i + 1) * layer_weights_num]
372 cu_weights.extend(self._tf_to_cudnn_weights(i, *layer_weights))
375 fw_weights, bw_weights = layer_weights[:len(
376 layer_weights) // 2], layer_weights[len(layer_weights) // 2:]
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | distributed_training_utils.py | 62 layer_weights = weights[:num_param]
63 for sw, w in zip(layer.weights, layer_weights):
|
| D | network.py | 470 layer_weights = weights[:num_param]
471 for sw, w in zip(layer.weights, layer_weights):
|