HomeSort by relevance Sort by last modified time
    Searched refs:layer_weights (Results 1 - 5 of 5) sorted by null

  /external/libaom/libaom/av1/encoder/
ml.c 31 const float *layer_weights = nn_config->weights[layer]; local
39 val += layer_weights[node * num_input_nodes + i] * input_nodes[i];
50 const float *layer_weights = nn_config->weights[num_layers]; local
55 val += layer_weights[node * num_input_nodes + i] * input_nodes[i];
  /external/libaom/libaom/av1/encoder/x86/
ml_sse3.c 162 const float *layer_weights = nn_config->weights[layer]; local
175 &layer_weights[out * num_inputs + in], &out_h,
187 &layer_weights[out * num_inputs + in], &outputs,
198 &layer_weights[out * num_inputs + in], &outputs,
209 &layer_weights[out * num_inputs + in], &total);
219 &layer_weights[out * num_inputs + in], &total);
232 _mm_load1_ps(&layer_weights[num_inputs * out + in_node]);
  /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/ops/
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:
    [all...]
  /external/tensorflow/tensorflow/python/keras/engine/
distributed_training_utils.py 62 layer_weights = weights[:num_param]
63 for sw, w in zip(layer.weights, layer_weights):
    [all...]
network.py 470 layer_weights = weights[:num_param]
471 for sw, w in zip(layer.weights, layer_weights):
    [all...]

Completed in 277 milliseconds