Searched refs:hidden_weights (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/framework/experimental/ |
| D | unified_api_test.py | 410 def model(x, hidden_weights, softmax_weight, labels): argument
412 for weight in hidden_weights + [softmax_weight]:
414 for hidden_weight in hidden_weights:
421 grads = tape.gradient(loss, hidden_weights + [softmax_weight])
425 hidden_weights = []
427 hidden_weights.append(
438 model(x, hidden_weights, softmax_weight, labels)
440 lambda: model(x, hidden_weights, softmax_weight, labels),
|
| /external/tensorflow/tensorflow/python/debug/examples/v2/ |
| D | debug_mnist_v2.py | 183 hidden_weights = get_dense_weights(IMAGE_SIZE**2, HIDDEN_SIZE)
185 variables = hidden_weights + output_weights
198 hidden_act = dense_layer(hidden_weights, x)
|
| /external/libtextclassifier/native/lang_id/common/ |
| D | embedding-network-params.h | 123 matrix.elements = hidden_weights(i); in GetHiddenLayerMatrix()
239 virtual const void *hidden_weights(int i) const = 0;
|
| /external/libtextclassifier/native/lang_id/common/flatbuffers/ |
| D | embedding-network-params-from-flatbuffer.h | 107 const void *hidden_weights(int i) const override { in hidden_weights() function
|