| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 42 inline bool ProcessBroadcastShapes(const RuntimeShape& shape0, in ProcessBroadcastShapes()
43 const RuntimeShape& shape1, in ProcessBroadcastShapes()
49 RuntimeShape scalar_shape(dims_count, 1); in ProcessBroadcastShapes()
51 auto extended_shape0 = RuntimeShape::ExtendedShape(dims_count, shape0); in ProcessBroadcastShapes()
52 auto extended_shape1 = RuntimeShape::ExtendedShape(dims_count, shape1); in ProcessBroadcastShapes()
88 const RuntimeShape* shape_a = in ProcessBroadcastShapes()
90 const RuntimeShape* shape_b = in ProcessBroadcastShapes()
133 inline void Conv(const ConvParams& params, const RuntimeShape& input_shape, in Conv()
134 const float* input_data, const RuntimeShape& filter_shape, in Conv()
135 const float* filter_data, const RuntimeShape& bias_shape, in Conv()
[all …]
|
| D | fully_connected.h | 30 const FullyConnectedParams& params, const RuntimeShape& input_shape, in FullyConnected()
31 const float* input_data, const RuntimeShape& weights_shape, in FullyConnected()
32 const float* weights_data, const RuntimeShape& bias_shape, in FullyConnected()
33 const float* bias_data, const RuntimeShape& output_shape, in FullyConnected()
66 const FullyConnectedParams& params, const RuntimeShape& input_shape, in FullyConnected()
67 const uint8* input_data, const RuntimeShape& filter_shape, in FullyConnected()
68 const uint8* filter_data, const RuntimeShape& bias_shape, in FullyConnected()
69 const int32* bias_data, const RuntimeShape& output_shape, in FullyConnected()
115 const FullyConnectedParams& params, const RuntimeShape& input_shape, in FullyConnected()
116 const uint8* input_data, const RuntimeShape& filter_shape, in FullyConnected()
[all …]
|
| D | depthwiseconv_uint8.h | 93 const RuntimeShape& input_shape, in Run()
95 const RuntimeShape& filter_shape, in Run()
97 const RuntimeShape& bias_shape, const int32* bias_data, in Run()
98 const RuntimeShape& output_shape, uint8* output_data) { in Run()
179 const DepthwiseParams& params, const RuntimeShape& input_shape, in DepthwiseConv()
180 const uint8* input_data, const RuntimeShape& filter_shape, in DepthwiseConv()
181 const uint8* filter_data, const RuntimeShape& bias_shape, in DepthwiseConv()
182 const int32* bias_data, const RuntimeShape& output_shape, in DepthwiseConv()
|
| D | depthwiseconv_float.h | 26 const DepthwiseParams& params, const RuntimeShape& input_shape, in DepthwiseConv()
27 const float* input_data, const RuntimeShape& filter_shape, in DepthwiseConv()
28 const float* filter_data, const RuntimeShape& bias_shape, in DepthwiseConv()
29 const float* bias_data, const RuntimeShape& output_shape, in DepthwiseConv()
|
| D | softmax.h | 29 const RuntimeShape& input_shape, const float* input_data, in Softmax()
30 const RuntimeShape& output_shape, float* output_data) { in Softmax()
61 const RuntimeShape& input_shape, const uint8* input_data, in Softmax()
62 const RuntimeShape& output_shape, uint8* output_data) { in Softmax()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | resize_nearest_neighbor_test.cc | 30 const RuntimeShape& input_shape, const std::vector<T>& input_data, in TestReferenceResizeNearestNeighbor()
32 const RuntimeShape& output_shape, in TestReferenceResizeNearestNeighbor()
35 RuntimeShape output_size_shape({1, 1, 1, 2}); in TestReferenceResizeNearestNeighbor()
48 RuntimeShape input_shape = {1, 2, 2, 1}; in TEST()
51 RuntimeShape output_shape = {1, 1, 1, 1}; in TEST()
59 RuntimeShape input_shape = {1, 2, 2, 1}; in TEST()
62 RuntimeShape output_shape = {1, 3, 3, 1}; in TEST()
70 RuntimeShape input_shape = {1, 3, 3, 1}; in TEST()
73 RuntimeShape output_shape = {1, 2, 2, 1}; in TEST()
81 RuntimeShape input_shape = {1, 2, 2, 1}; in TEST()
[all …]
|
| D | types.h | 121 class RuntimeShape {
127 RuntimeShape& operator=(RuntimeShape const&) = delete;
129 RuntimeShape() : size_(0) {} in RuntimeShape() function
131 explicit RuntimeShape(int dimensions_count) : size_(dimensions_count) { in RuntimeShape() function
141 RuntimeShape(int shape_size, int32 value) : size_(0) { in RuntimeShape() function
148 RuntimeShape(int dimensions_count, const int32* dims_data) : size_(0) { in RuntimeShape() function
152 RuntimeShape(const std::initializer_list<int> init_list) : size_(0) { in RuntimeShape() function
158 RuntimeShape(RuntimeShape const& other) : size_(other.DimensionsCount()) { in RuntimeShape() function
165 bool operator==(const RuntimeShape& comp) const {
170 ~RuntimeShape() { in ~RuntimeShape()
[all …]
|
| D | depthwiseconv_float_test.cc | 33 const DepthwiseParams& params, const RuntimeShape& input_shape, in TestOneDepthwiseConv()
34 const float* input_data, const RuntimeShape& filter_shape, in TestOneDepthwiseConv()
35 const float* filter_data, const RuntimeShape& bias_shape, in TestOneDepthwiseConv()
36 const float* bias_data, const RuntimeShape& output_shape) { in TestOneDepthwiseConv()
100 RuntimeShape input_shape_inference( in TryTestOneDepthwiseConv()
102 RuntimeShape output_shape_inference; in TryTestOneDepthwiseConv()
112 RuntimeShape filter_shape_inference( in TryTestOneDepthwiseConv()
114 RuntimeShape bias_shape_inference({1, 1, 1, output_depth}); in TryTestOneDepthwiseConv()
|
| D | depthwiseconv_quantized_test.cc | 107 const RuntimeShape& input_shape, const uint8* input_data, in DispatchDepthwiseConv()
108 const RuntimeShape& filter_shape, const uint8* filter_data, in DispatchDepthwiseConv()
109 const RuntimeShape& bias_shape, const int32* bias_data, in DispatchDepthwiseConv()
110 const RuntimeShape& output_shape, uint8* output_data) { in DispatchDepthwiseConv()
290 const RuntimeShape& input_shape, std::int32_t input_offset, in TestOneDepthwiseConvWithGivenOutputShift()
291 const std::uint8_t* filter_data, const RuntimeShape& filter_shape, in TestOneDepthwiseConvWithGivenOutputShift()
293 const RuntimeShape& bias_shape, int stride, PaddingType padding_type, in TestOneDepthwiseConvWithGivenOutputShift()
297 std::int32_t output_activation_max, const RuntimeShape& output_shape) { in TestOneDepthwiseConvWithGivenOutputShift()
423 const RuntimeShape& input_shape, std::int32_t input_offset, in TestOneDepthwiseConvBisectOutputShift()
424 const std::uint8_t* filter_data, const RuntimeShape& filter_shape, in TestOneDepthwiseConvBisectOutputShift()
[all …]
|
| D | tensor_test.cc | 25 RuntimeShape d = GetTensorShape({2, 3, 4, 5}); in TEST()
32 RuntimeShape d = GetTensorShape({3, 4, 5}); in TEST()
39 RuntimeShape d = GetTensorShape({4, 5}); in TEST()
46 RuntimeShape d = GetTensorShape({5}); in TEST()
|
| D | tensor.h | 40 inline RuntimeShape GetTensorShape(std::vector<int32_t> data) { in GetTensorShape()
41 return RuntimeShape(data.size(), data.data()); in GetTensorShape()
81 const RuntimeShape* const* shapes() const { return all_shape_ptr_.data(); } in shapes()
85 std::vector<RuntimeShape> all_shape_;
86 std::vector<RuntimeShape*> all_shape_ptr_;
|
| D | logsoftmax_quantized_test.cc | 38 const RuntimeShape& shape_common, in RunLogSoftmaxFloatReference()
71 const RuntimeShape& shape_common, in RunLogSoftmaxFloatReference()
102 const RuntimeShape& shape_common, in CheckOutputData()
147 const RuntimeShape& shape_common, int32 input_offset, in RunOneLogSoftmaxTest()
199 const RuntimeShape& shape_common, int32 input_offset, in RunOneLogSoftmaxTest()
259 RuntimeShape({batch, input_height, input_width, input_depth}); in TryOneUniformLogSoftmax()
297 RuntimeShape({batch, input_height, input_width, input_depth}); in TryOneSkyscraperLogSoftmax()
|
| D | softmax_quantized_test.cc | 36 const RuntimeShape& shape_common, in RunSoftmaxFloatReference()
65 const RuntimeShape& shape_common, in CheckOutputData()
102 const RuntimeShape& shape_common, int32 input_offset, in RunOneSoftmaxTest()
163 RuntimeShape({batch, input_height, input_width, input_depth}); in TryOneUniformSoftmax()
201 RuntimeShape({batch, input_height, input_width, input_depth}); in TryOneSkyscraperSoftmax()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/ |
| D | types.h | 122 class RuntimeShape {
128 RuntimeShape& operator=(RuntimeShape const&) = delete;
130 RuntimeShape() : size_(0) {} in RuntimeShape() function
132 explicit RuntimeShape(int dimensions_count) : size_(dimensions_count) { in RuntimeShape() function
138 RuntimeShape(int shape_size, int32 value) : size_(0) { in RuntimeShape() function
145 RuntimeShape(int dimensions_count, const int32* dims_data) : size_(0) { in RuntimeShape() function
149 RuntimeShape(const std::initializer_list<int> init_list) : size_(0) { in RuntimeShape() function
155 RuntimeShape(RuntimeShape const& other) : size_(other.DimensionsCount()) { in RuntimeShape() function
162 bool operator==(const RuntimeShape& comp) const {
167 ~RuntimeShape() { in ~RuntimeShape()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 115 VectorMap<Scalar> MapAsVector(Scalar* data, const RuntimeShape& shape) { in MapAsVector()
132 const RuntimeShape& shape) { in MapAsMatrixWithLastDimAsRows()
141 const RuntimeShape& shape) { in MapAsMatrixWithFirstDimAsCols()
156 const RuntimeShape& shape) { in MapAsArrayWithLastDimAsRows()
179 const RuntimeShape& shape, in MapAsMatrixWithGivenNumberOfRows()
189 const RuntimeShape& bias_shape, in AddBiasAndEvalActivationFunction()
191 const RuntimeShape& array_shape, in AddBiasAndEvalActivationFunction()
293 inline void GEMVForLstmCell(const RuntimeShape& input_shape, in GEMVForLstmCell()
295 const RuntimeShape& weights_shape, in GEMVForLstmCell()
297 const RuntimeShape& bias_shape, in GEMVForLstmCell()
[all …]
|
| D | multithreaded_conv.h | 136 const ConvParams& params, const RuntimeShape& input_shape, in Conv()
137 const float* input_data, const RuntimeShape& filter_shape, in Conv()
138 const float* filter_data, const RuntimeShape& bias_shape, in Conv()
139 const float* bias_data, const RuntimeShape& output_shape, in Conv()
140 float* output_data, const RuntimeShape& im2col_shape, in Conv()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 109 VectorMap<Scalar> MapAsVector(Scalar* data, const RuntimeShape& shape) { in MapAsVector()
132 const RuntimeShape& shape) { in MapAsMatrixWithLastDimAsRows()
141 const RuntimeShape& shape) { in MapAsMatrixWithFirstDimAsCols()
189 const RuntimeShape& shape) { in MapAsArrayWithLastDimAsRows()
226 const RuntimeShape& shape, in MapAsMatrixWithGivenNumberOfRows()
284 const RuntimeShape& bias_shape, in AddBiasAndEvalActivationFunction()
286 const RuntimeShape& array_shape, in AddBiasAndEvalActivationFunction()
415 inline void GEMVForLstmCell(const RuntimeShape& input_shape, in GEMVForLstmCell()
417 const RuntimeShape& weights_shape, in GEMVForLstmCell()
419 const RuntimeShape& bias_shape, in GEMVForLstmCell()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/ |
| D | mul.h | 43 const RuntimeShape& input1_shape, const int8_t* input1_data, in Mul()
44 const RuntimeShape& input2_shape, const int8_t* input2_data, in Mul()
45 const RuntimeShape& output_shape, int8_t* output_data) { in Mul()
57 const RuntimeShape& input1_shape, const int16* input1_data, in Mul()
58 const RuntimeShape& input2_shape, const int16* input2_data, in Mul()
59 const RuntimeShape& output_shape, int8_t* output_data) { in Mul()
86 const RuntimeShape& input1_shape, in BroadcastMul4DSlow()
88 const RuntimeShape& input2_shape, in BroadcastMul4DSlow()
90 const RuntimeShape& output_shape, in BroadcastMul4DSlow()
99 const RuntimeShape extended_output_shape = in BroadcastMul4DSlow()
[all …]
|
| D | add.h | 61 const RuntimeShape& input1_shape, const int8_t* input1_data, in Add()
62 const RuntimeShape& input2_shape, const int8_t* input2_data, in Add()
63 const RuntimeShape& output_shape, int8_t* output_data) { in Add()
78 const RuntimeShape& input1_shape, in BroadcastAdd4DSlow()
80 const RuntimeShape& input2_shape, in BroadcastAdd4DSlow()
82 const RuntimeShape& output_shape, in BroadcastAdd4DSlow()
89 const RuntimeShape extended_output_shape = in BroadcastAdd4DSlow()
90 RuntimeShape::ExtendedShape(4, output_shape); in BroadcastAdd4DSlow()
|
| D | mean.h | 24 int32_t shift, const RuntimeShape& unextended_input_shape, in Mean()
26 const RuntimeShape& unextended_output_shape, in Mean()
32 const RuntimeShape input_shape = in Mean()
33 RuntimeShape::ExtendedShape(4, unextended_input_shape); in Mean()
34 const RuntimeShape output_shape = in Mean()
35 RuntimeShape::ExtendedShape(4, unextended_output_shape); in Mean()
|
| D | fully_connected.h | 24 const FullyConnectedParams& params, const RuntimeShape& input_shape, in FullyConnected()
25 const int8_t* input_data, const RuntimeShape& filter_shape, in FullyConnected()
26 const int8_t* filter_data, const RuntimeShape& bias_shape, in FullyConnected()
27 const int32* bias_data, const RuntimeShape& output_shape, in FullyConnected()
|
| D | conv.h | 26 const int32* output_shift, const RuntimeShape& input_shape, in ConvPerChannel()
27 const int8* input_data, const RuntimeShape& filter_shape, in ConvPerChannel()
28 const int8* filter_data, const RuntimeShape& bias_shape, in ConvPerChannel()
29 const int32* bias_data, const RuntimeShape& output_shape, in ConvPerChannel()
|
| D | depthwise_conv.h | 24 const int32* output_shift, const RuntimeShape& input_shape, in DepthwiseConvPerChannel()
25 const int8* input_data, const RuntimeShape& filter_shape, in DepthwiseConvPerChannel()
26 const int8* filter_data, const RuntimeShape& bias_shape, in DepthwiseConvPerChannel()
27 const int32* bias_data, const RuntimeShape& output_shape, in DepthwiseConvPerChannel()
|
| D | pooling.h | 25 const RuntimeShape& input_shape, const int8* input_data, in AveragePool()
26 const RuntimeShape& output_shape, int8* output_data) { in AveragePool()
81 inline void MaxPool(const PoolParams& params, const RuntimeShape& input_shape, in MaxPool()
82 const int8* input_data, const RuntimeShape& output_shape, in MaxPool()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | depthwise_conv.cc | 197 void (*depthwise_conv)(const DepthwiseParams&, const RuntimeShape&, in EvalFloat()
198 const float*, const RuntimeShape&, const float*, in EvalFloat()
199 const RuntimeShape&, const float*, const RuntimeShape&, in EvalFloat()
233 void (*depthwise_conv)(const DepthwiseParams&, const RuntimeShape&, in EvalQuantized()
234 const uint8*, const RuntimeShape&, const uint8*, in EvalQuantized()
235 const RuntimeShape&, const int32*, const RuntimeShape&, in EvalQuantized()
|