Searched refs:output_matrix_size_bytes (Results 1 – 3 of 3) sorted by relevance
| /external/ComputeLibrary/src/core/NEON/kernels/assembly/ |
| D | winograd.hpp | 79 size_t weight_matrix_size_bytes, input_matrix_size_bytes, output_matrix_size_bytes; member
|
| /external/ComputeLibrary/src/cpu/operators/ |
| D | CpuWinogradConv2d.cpp | 226 d_info.init(d_shape, 1, data_type, d_strides, 0, wds.output_matrix_size_bytes); in configure()
280 …fset_int_vec(TransformedOutput), MemoryLifetime::Temporary, wds.output_matrix_size_bytes, storage_… in configure()
|
| /external/ComputeLibrary/src/core/NEON/kernels/convolution/winograd/ |
| D | winograd_implementations.hpp | 335 ws.output_matrix_size_bytes = conv_args.n_batches * ws.output_ld_batch * sizeof(TWinogradOut); in get_implementation()
|