| /frameworks/ml/nn/common/operations/internal/reference/ |
| D | reference_ops.h | 164 const Dims<4>& output_dims, float* im2col_data, in Conv() argument
168 const int batches = MatchingArraySize(input_dims, 3, output_dims, 3); in Conv()
170 const int output_depth = MatchingArraySize(filter_dims, 3, output_dims, 0); in Conv()
178 const int output_height = ArraySize(output_dims, 2); in Conv()
179 const int output_width = ArraySize(output_dims, 1); in Conv()
210 output_data[Offset(output_dims, out_channel, out_x, out_y, batch)] = in Conv()
226 uint8* output_data, const Dims<4>& output_dims, uint8* im2col_data, in Conv() argument
241 const int batches = MatchingArraySize(input_dims, 3, output_dims, 3); in Conv()
244 MatchingArraySize(filter_dims, 3, bias_dims, 0, output_dims, 0); in Conv()
249 const int output_height = ArraySize(output_dims, 2); in Conv()
[all …]
|
| D | depthwiseconv_float.h | 33 float* output_data, const Dims<4>& output_dims) { in DepthwiseConv() argument
34 const int batches = MatchingArraySize(input_dims, 3, output_dims, 3); in DepthwiseConv()
35 const int output_depth = MatchingArraySize(filter_dims, 0, output_dims, 0); in DepthwiseConv()
41 const int output_height = ArraySize(output_dims, 2); in DepthwiseConv()
42 const int output_width = ArraySize(output_dims, 1); in DepthwiseConv()
74 output_data[Offset(output_dims, oc, out_x, out_y, b)] = in DepthwiseConv()
|
| D | depthwiseconv_uint8.h | 39 const Dims<4>& output_dims) { in DepthwiseConv() argument
50 const int batches = MatchingArraySize(input_dims, 3, output_dims, 3); in DepthwiseConv()
51 const int output_depth = MatchingArraySize(filter_dims, 0, output_dims, 0); in DepthwiseConv()
57 const int output_height = ArraySize(output_dims, 2); in DepthwiseConv()
58 const int output_width = ArraySize(output_dims, 1); in DepthwiseConv()
95 output_data[Offset(output_dims, oc, out_x, out_y, b)] = in DepthwiseConv()
|
| /frameworks/ml/nn/common/operations/internal/optimized/ |
| D | optimized_ops.h | 336 float* output_data, const Dims<4>& output_dims) { in FullyConnected() argument
352 MapAsMatrixWithFirstDimAsRows(output_data, output_dims); in FullyConnected()
356 output_dims); in FullyConnected()
376 const Dims<4>& output_dims) { in FullyConnectedAsGEMV() argument
386 DCHECK(IsPackedWithoutStrides(output_dims)); in FullyConnectedAsGEMV()
387 DCHECK_EQ(ArraySize(output_dims, 1) * ArraySize(output_dims, 2) * in FullyConnectedAsGEMV()
388 ArraySize(output_dims, 3), in FullyConnectedAsGEMV()
391 const int output_size = MatchingArraySize(filter_dims, 1, output_dims, 0); in FullyConnectedAsGEMV()
596 const Dims<4>& output_dims,
609 const int batches = ArraySize(output_dims, 1) * ArraySize(output_dims, 2) *
[all …]
|
| D | depthwiseconv_float.h | 619 float* output_data, const Dims<4>& output_dims) {
626 const int batches = MatchingArraySize(input_dims, 3, output_dims, 3);
627 const int output_depth = MatchingArraySize(filter_dims, 0, output_dims, 0);
633 const int output_height = ArraySize(output_dims, 2);
634 const int output_width = ArraySize(output_dims, 1);
|
| D | depthwiseconv_uint8.h | 1354 const Dims<4>& output_dims) {
1366 const int batches = MatchingArraySize(input_dims, 3, output_dims, 3);
1367 const int output_depth = MatchingArraySize(filter_dims, 0, output_dims, 0);
1373 const int output_height = ArraySize(output_dims, 2);
1374 const int output_width = ArraySize(output_dims, 1);
|