| /frameworks/ml/nn/common/operations/internal/reference/ |
| D | reference_ops.h | 160 void Conv(const float* input_data, const Dims<4>& input_dims, in Conv() argument
168 const int batches = MatchingArraySize(input_dims, 3, output_dims, 3); in Conv()
169 const int input_depth = MatchingArraySize(input_dims, 0, filter_dims, 0); in Conv()
174 const int input_height = ArraySize(input_dims, 2); in Conv()
175 const int input_width = ArraySize(input_dims, 1); in Conv()
196 float input_value = input_data[Offset(input_dims, in_channel, in Conv()
219 void Conv(const uint8* input_data, const Dims<4>& input_dims, in Conv() argument
241 const int batches = MatchingArraySize(input_dims, 3, output_dims, 3); in Conv()
242 const int input_depth = MatchingArraySize(input_dims, 0, filter_dims, 0); in Conv()
245 const int input_height = ArraySize(input_dims, 2); in Conv()
[all …]
|
| D | depthwiseconv_float.h | 28 void DepthwiseConv(const float* input_data, const Dims<4>& input_dims, in DepthwiseConv() argument
34 const int batches = MatchingArraySize(input_dims, 3, output_dims, 3); in DepthwiseConv()
36 const int input_height = ArraySize(input_dims, 2); in DepthwiseConv()
37 const int input_width = ArraySize(input_dims, 1); in DepthwiseConv()
38 const int input_depth = ArraySize(input_dims, 0); in DepthwiseConv()
63 input_data[Offset(input_dims, ic, in_x, in_y, b)]; in DepthwiseConv()
|
| D | depthwiseconv_uint8.h | 30 void DepthwiseConv(const uint8* input_data, const Dims<4>& input_dims, in DepthwiseConv() argument
50 const int batches = MatchingArraySize(input_dims, 3, output_dims, 3); in DepthwiseConv()
52 const int input_height = ArraySize(input_dims, 2); in DepthwiseConv()
53 const int input_width = ArraySize(input_dims, 1); in DepthwiseConv()
54 const int input_depth = ArraySize(input_dims, 0); in DepthwiseConv()
79 input_data[Offset(input_dims, ic, in_x, in_y, b)]; in DepthwiseConv()
|
| /frameworks/ml/nn/common/operations/internal/optimized/ |
| D | optimized_ops.h | 333 void FullyConnected(const float* input_data, const Dims<4>& input_dims, in FullyConnected() argument
348 MapAsMatrixWithGivenNumberOfRows(input_data, input_dims, input_rows); in FullyConnected()
369 void FullyConnectedAsGEMV(const uint8* input_data, const Dims<4>& input_dims, in FullyConnectedAsGEMV() argument
383 DCHECK(IsPackedWithoutStrides(input_dims)); in FullyConnectedAsGEMV()
390 const int input_size = input_dims.strides[3]; in FullyConnectedAsGEMV()
589 void FullyConnected(const uint8* input_data, const Dims<4>& input_dims,
615 input_data, input_dims, input_offset, filter_data, filter_dims,
649 const Dims<4>& input_dims, int w, int h, int b, int kheight, int kwidth,
675 int in_offset = Offset(input_dims, 0, iw_start, ih_start, b);
734 void Im2col(const T* input_data, const Dims<4>& input_dims, int stride_width,
[all …]
|
| D | depthwiseconv_float.h | 614 void DepthwiseConv(const float* input_data, const Dims<4>& input_dims,
626 const int batches = MatchingArraySize(input_dims, 3, output_dims, 3);
628 const int input_height = ArraySize(input_dims, 2);
629 const int input_width = ArraySize(input_dims, 1);
630 const int input_depth = ArraySize(input_dims, 0);
715 input_data + in_y * input_dims.strides[2] +
716 b * input_dims.strides[3],
|
| D | depthwiseconv_uint8.h | 1345 void DepthwiseConv(const uint8* input_data, const Dims<4>& input_dims,
1366 const int batches = MatchingArraySize(input_dims, 3, output_dims, 3);
1368 const int input_height = ArraySize(input_dims, 2);
1369 const int input_width = ArraySize(input_dims, 1);
1370 const int input_depth = ArraySize(input_dims, 0);
1465 input_data + in_y * input_dims.strides[2] +
1466 b * input_dims.strides[3],
|