Searched refs:fully_connected_layer (Results 1 – 8 of 8) sorted by relevance
| /external/ComputeLibrary/tests/validation/reference/ |
| D | FullyConnectedLayer.cpp | 115 SimpleTensor<T> fully_connected_layer(const SimpleTensor<T> &src, const SimpleTensor<T> &weights, c… in fully_connected_layer() function
161 template SimpleTensor<float> fully_connected_layer(const SimpleTensor<float> &src, const SimpleTens…
163 template SimpleTensor<half> fully_connected_layer(const SimpleTensor<half> &src, const SimpleTensor…
165 template SimpleTensor<uint8_t> fully_connected_layer(const SimpleTensor<uint8_t> &src, const Simple…
167 template SimpleTensor<int8_t> fully_connected_layer(const SimpleTensor<int8_t> &src, const SimpleTe…
|
| D | FullyConnectedLayer.h | 39 SimpleTensor<T> fully_connected_layer(const SimpleTensor<T> &src, const SimpleTensor<T> &weights, c…
|
| /external/ComputeLibrary/tests/validation/fixtures/UNIT/ |
| D | MemoryManagerFixture.h | 136 auto fc1 = reference::fully_connected_layer(src, w1, b1, TensorShape(128U)); in compute_reference()
137 return reference::fully_connected_layer(fc1, w2, b2, TensorShape(24U)); in compute_reference()
272 auto fc1 = reference::fully_connected_layer(src, w1, b1, TensorShape(128U, _cur_batches)); in compute_reference()
273 return reference::fully_connected_layer(fc1, w2, b2, TensorShape(24U, _cur_batches)); in compute_reference()
399 auto fc = reference::fully_connected_layer(src, w, b, TensorShape(8U, _cur_batches)); in compute_reference()
|
| D | WeightsRetentionFixture.h | 141 return reference::fully_connected_layer(src, w1, b1, TensorShape(15U, _cur_batches)); in compute_reference()
|
| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | LSTMLayerFixture.h | 409 …SimpleTensor<T> fully_connected_forget = reference::fully_connected_layer(input, input_to_forget_w… in compute_reference()
441 …SimpleTensor<T> fully_connected_input = reference::fully_connected_layer(input, input_to_input_w, … in compute_reference()
462 …SimpleTensor<T> fully_connected_cell_state = reference::fully_connected_layer(input, input_to_cell… in compute_reference()
486 …SimpleTensor<T> fully_connected_output = reference::fully_connected_layer(input, input_to_output_w… in compute_reference()
512 …SimpleTensor<T> fully_connected_projection = reference::fully_connected_layer(output_state_out, pr… in compute_reference()
|
| D | RNNLayerFixture.h | 136 …SimpleTensor<T> fully_connected = reference::fully_connected_layer(input, weights, bias, out_shape… in compute_reference()
|
| D | FullyConnectedLayerFixture.h | 235 …return reference::activation_layer(reference::fully_connected_layer<T>(src, weights, bias, output_… in compute_reference()
454 …auto dst = reference::activation_layer(reference::fully_connected_layer<T>(src, weights, bias, dst… in setup()
|
| /external/ComputeLibrary/tests/validate_examples/ |
| D | graph_fully_connected.cpp | 202 …return reference::fully_connected_layer<D>(src, weights, bias, output_shape, _params.output.quant_… in reference()
|