Searched refs:concatenate_layer (Results 1 – 4 of 4) sorted by relevance
/aosp_15_r20/external/ComputeLibrary/tests/validation/reference/ |
H A D | ConcatenateLayer.cpp | 116 SimpleTensor<T> concatenate_layer(std::vector<SimpleTensor<T>> &srcs, SimpleTensor<T> &dst, unsigne… in concatenate_layer() function 160 template SimpleTensor<float> concatenate_layer(std::vector<SimpleTensor<float>> &srcs, SimpleTensor… 161 template SimpleTensor<half> concatenate_layer(std::vector<SimpleTensor<half>> &srcs, SimpleTensor<h… 162 template SimpleTensor<uint8_t> concatenate_layer(std::vector<SimpleTensor<uint8_t>> &srcs, SimpleTe… 163 template SimpleTensor<int8_t> concatenate_layer(std::vector<SimpleTensor<int8_t>> &srcs, SimpleTens…
|
H A D | ConcatenateLayer.h | 40 SimpleTensor<T> concatenate_layer(std::vector<SimpleTensor<T>> &srcs, SimpleTensor<T> &dst, unsigne…
|
/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/ |
H A D | ConcatenateLayerFixture.h | 165 return reference::concatenate_layer<T>(srcs, dst, axis); in compute_reference()
|
H A D | LSTMLayerFixture.h | 526 scratch = reference::concatenate_layer(scratch_inputs, scratch, Window::DimX); in compute_reference()
|