| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | transpose_utils_test.cc | 24 RuntimeShape input_shape({9}); in TEST()
25 RuntimeShape output_shape({9}); in TEST()
34 EXPECT_EQ(input_shape, RuntimeShape({9})); in TEST()
35 EXPECT_EQ(output_shape, RuntimeShape({9})); in TEST()
42 RuntimeShape input_shape({9, 3}); in TEST()
43 RuntimeShape output_shape({3, 9}); in TEST()
53 EXPECT_EQ(input_shape, RuntimeShape({9, 3})); in TEST()
54 EXPECT_EQ(output_shape, RuntimeShape({3, 9})); in TEST()
62 RuntimeShape input_shape({9, 1}); in TEST()
63 RuntimeShape output_shape({1, 9}); in TEST()
[all …]
|
| 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()
36 RuntimeShape output_size_shape({1, 1, 1, 2}); in TestReferenceResizeNearestNeighbor()
49 RuntimeShape input_shape = {1, 2, 2, 1}; in TEST()
52 RuntimeShape output_shape = {1, 1, 1, 1}; in TEST()
60 RuntimeShape input_shape = {1, 2, 2, 1}; in TEST()
63 RuntimeShape output_shape = {1, 1, 1, 1}; in TEST()
72 RuntimeShape input_shape = {1, 2, 2, 1}; in TEST()
75 RuntimeShape output_shape = {1, 1, 1, 1}; in TEST()
84 RuntimeShape input_shape = {1, 2, 2, 1}; in TEST()
[all …]
|
| D | types.h | 148 class RuntimeShape {
154 RuntimeShape& operator=(RuntimeShape const&) = delete;
156 RuntimeShape() : size_(0) {} in RuntimeShape() function
158 explicit RuntimeShape(int dimensions_count) : size_(dimensions_count) { in RuntimeShape() function
168 RuntimeShape(int shape_size, int32_t value) : size_(0) { in RuntimeShape() function
175 RuntimeShape(int dimensions_count, const int32_t* dims_data) : size_(0) { in RuntimeShape() function
179 RuntimeShape(const std::initializer_list<int> init_list) : size_(0) { in RuntimeShape() function
185 RuntimeShape(RuntimeShape const& other) : size_(other.DimensionsCount()) { in RuntimeShape() function
196 bool operator==(const RuntimeShape& comp) const {
202 ~RuntimeShape() { in ~RuntimeShape()
[all …]
|
| D | transpose_utils.h | 27 const RuntimeShape& input_shape, int* dim0,
32 void RemoveOneSizeDimensions(RuntimeShape* input_shape,
33 RuntimeShape* output_shape,
42 size_t Flatten(const RuntimeShape& input_shape,
43 const RuntimeShape& output_shape, const TransposeParams& params,
44 RuntimeShape* non_flatten_input_shape,
45 RuntimeShape* non_flatten_output_shape,
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | comparisons.h | 58 const ComparisonParams& op_params, const RuntimeShape& input1_shape, in ComparisonImpl()
59 const T* input1_data, const RuntimeShape& input2_shape, in ComparisonImpl()
60 const T* input2_data, const RuntimeShape& output_shape, bool* output_data) { in ComparisonImpl()
70 const RuntimeShape& input1_shape, in Comparison()
72 const RuntimeShape& input2_shape, in Comparison()
74 const RuntimeShape& output_shape, bool* output_data) { in Comparison()
81 const ComparisonParams& op_params, const RuntimeShape& input1_shape, in ComparisonWithScaling()
82 const T* input1_data, const RuntimeShape& input2_shape, in ComparisonWithScaling()
83 const T* input2_data, const RuntimeShape& output_shape, bool* output_data) { in ComparisonWithScaling()
110 const RuntimeShape output_shape;
[all …]
|
| D | reference_ops.h | 91 inline void Relu(const RuntimeShape& input_shape, const T* input_data, in Relu()
92 const RuntimeShape& output_shape, T* output_data) { in Relu()
103 inline void Relu1(const RuntimeShape& input_shape, const T* input_data, in Relu1()
104 const RuntimeShape& output_shape, T* output_data) { in Relu1()
116 inline void Relu6(const RuntimeShape& input_shape, const float* input_data, in Relu6()
117 const RuntimeShape& output_shape, float* output_data) { in Relu6()
131 const RuntimeShape& input_shape, const T* input_data, in ReluX()
132 const RuntimeShape& output_shape, T* output_data) { in ReluX()
149 const RuntimeShape& input_shape, const T* input_data, in ReluX()
150 const RuntimeShape& output_shape, T* output_data) { in ReluX()
[all …]
|
| D | sub.h | 33 const RuntimeShape& input1_shape, in SubNonBroadcast()
35 const RuntimeShape& input2_shape, in SubNonBroadcast()
37 const RuntimeShape& output_shape, in SubNonBroadcast()
49 const RuntimeShape& input1_shape, in SubNonBroadcast()
51 const RuntimeShape& input2_shape, in SubNonBroadcast()
53 const RuntimeShape& output_shape, in SubNonBroadcast()
70 const RuntimeShape& input1_shape, in BroadcastSubSlow()
72 const RuntimeShape& input2_shape, in BroadcastSubSlow()
74 const RuntimeShape& output_shape, in BroadcastSubSlow()
85 CopyDimsToDesc(RuntimeShape::ExtendedShape(N, output_shape), &output_desc); in BroadcastSubSlow()
[all …]
|
| D | pad.h | 39 const RuntimeShape& input_shape, const T* input_data, in PadImpl()
40 const P* pad_value_ptr, const RuntimeShape& output_shape, in PadImpl()
42 const RuntimeShape ext_input_shape = in PadImpl()
43 RuntimeShape::ExtendedShape(PadKernelMaxDimensionCount(), input_shape); in PadImpl()
44 const RuntimeShape ext_output_shape = in PadImpl()
45 RuntimeShape::ExtendedShape(PadKernelMaxDimensionCount(), output_shape); in PadImpl()
112 const RuntimeShape& input_shape, const T* input_data, in Pad()
113 const P* pad_value_ptr, const RuntimeShape& output_shape, in Pad()
122 const RuntimeShape& input_shape, const T* input_data, in Pad()
123 const int32_t* pad_value_ptr, const RuntimeShape& output_shape, in Pad()
[all …]
|
| D | mul.h | 46 const RuntimeShape& input1_shape, const T* input1_data, in Mul()
47 const RuntimeShape& input2_shape, const T* input2_data, in Mul()
48 const RuntimeShape& output_shape, T* output_data) { in Mul()
63 const RuntimeShape& input1_shape, const uint8_t* input1_data, in Mul()
64 const RuntimeShape& input2_shape, const uint8_t* input2_data, in Mul()
65 const RuntimeShape& output_shape, uint8_t* output_data) { in Mul()
75 const RuntimeShape& input1_shape, in BroadcastMul4DSlow()
77 const RuntimeShape& input2_shape, in BroadcastMul4DSlow()
79 const RuntimeShape& output_shape, in BroadcastMul4DSlow()
85 const RuntimeShape extended_output_shape = in BroadcastMul4DSlow()
[all …]
|
| D | add.h | 27 const RuntimeShape& input1_shape, const T* input1_data, in Add()
28 const RuntimeShape& input2_shape, const T* input2_data, in Add()
29 const RuntimeShape& output_shape, T* output_data) { in Add()
40 const RuntimeShape& input1_shape, const float* input1_data, in Add()
41 const RuntimeShape& input2_shape, const float* input2_data, in Add()
42 const RuntimeShape& output_shape, float* output_data) { in Add()
125 const RuntimeShape& input1_shape, const uint8_t* input1_data, in Add()
126 const RuntimeShape& input2_shape, const uint8_t* input2_data, in Add()
127 const RuntimeShape& output_shape, uint8_t* output_data) { in Add()
141 const RuntimeShape& input1_shape, in AddGeneralParamScale()
[all …]
|
| D | batch_matmul.h | 40 inline int extent(const RuntimeShape& shape, int x) { in extent()
53 inline void BatchMatMul(const RuntimeShape& lhs_shape, const float* lhs_data, in BatchMatMul()
54 const RuntimeShape& rhs_shape, const float* rhs_data, in BatchMatMul()
55 const RuntimeShape& output_shape, float* output_data) { in BatchMatMul()
56 const RuntimeShape extended_lhs_shape = in BatchMatMul()
57 RuntimeShape::ExtendedShape(5, lhs_shape); in BatchMatMul()
58 const RuntimeShape extended_rhs_shape = in BatchMatMul()
59 RuntimeShape::ExtendedShape(5, rhs_shape); in BatchMatMul()
108 inline void BatchMatMul(const RuntimeShape& lhs_shape, const int8_t* lhs_data, in BatchMatMul()
109 const RuntimeShape& rhs_shape, const int8_t* rhs_data, in BatchMatMul()
[all …]
|
| D | batch_to_space_nd.h | 27 inline RuntimeShape ExtendShapeBatchToSpace(const RuntimeShape& shape) { in ExtendShapeBatchToSpace()
31 RuntimeShape new_shape(4, 1); in ExtendShapeBatchToSpace()
39 inline void BatchToSpaceND(const RuntimeShape& unextended_input1_shape, in BatchToSpaceND()
41 const RuntimeShape& unextended_input2_shape, in BatchToSpaceND()
43 const RuntimeShape& unextended_input3_shape, in BatchToSpaceND()
45 const RuntimeShape& unextended_output_shape, in BatchToSpaceND()
53 const RuntimeShape input1_shape = in BatchToSpaceND()
55 const RuntimeShape output_shape = in BatchToSpaceND()
|
| D | div.h | 62 const RuntimeShape& input1_shape, const uint8* input1_data, in Div()
63 const RuntimeShape& input2_shape, const uint8* input2_data, in Div()
64 const RuntimeShape& output_shape, uint8* output_data) { in Div()
75 const RuntimeShape& unextended_input1_shape, in BroadcastDivSlow()
77 const RuntimeShape& unextended_input2_shape, in BroadcastDivSlow()
79 const RuntimeShape& unextended_output_shape, in BroadcastDivSlow()
90 CopyDimsToDesc(RuntimeShape::ExtendedShape(N, unextended_output_shape), in BroadcastDivSlow()
133 const RuntimeShape& unextended_input1_shape, in BroadcastDivSlow()
135 const RuntimeShape& unextended_input2_shape, in BroadcastDivSlow()
137 const RuntimeShape& unextended_output_shape, in BroadcastDivSlow()
[all …]
|
| D | space_to_batch_nd.h | 28 inline RuntimeShape ExtendShapeSpaceToBatch(const RuntimeShape& shape) { in ExtendShapeSpaceToBatch()
32 RuntimeShape new_shape(4, 1); in ExtendShapeSpaceToBatch()
41 const RuntimeShape& unextended_input1_shape, in SpaceToBatchND()
43 const RuntimeShape& unextended_input2_shape, in SpaceToBatchND()
45 const RuntimeShape& unextended_input3_shape, in SpaceToBatchND()
47 const RuntimeShape& unextended_output_shape, in SpaceToBatchND()
56 const RuntimeShape input1_shape = in SpaceToBatchND()
58 const RuntimeShape output_shape = in SpaceToBatchND()
|
| D | svdf.h | 67 const TfLiteSVDFParams* params, const RuntimeShape& input_shape, in EvalIntegerSVDF()
68 const int8_t* input_data, const RuntimeShape& weights_feature_shape, in EvalIntegerSVDF()
69 const int8_t* weights_feature_data, const RuntimeShape& weights_time_shape, in EvalIntegerSVDF()
70 const int16_t* weights_time_data, const RuntimeShape& bias_shape, in EvalIntegerSVDF()
72 const RuntimeShape& output_shape, int8_t* output_data, in EvalIntegerSVDF()
153 const TfLiteSVDFParams* params, const RuntimeShape& input_shape, in EvalFloatSVDF()
154 const float* input_data, const RuntimeShape& weights_feature_shape, in EvalFloatSVDF()
155 const float* weights_feature_data, const RuntimeShape& weights_time_shape, in EvalFloatSVDF()
156 const float* weights_time_data, const RuntimeShape& bias_shape, in EvalFloatSVDF()
158 const RuntimeShape& output_shape, float* output_data) { in EvalFloatSVDF()
[all …]
|
| /external/tensorflow/tensorflow/lite/experimental/kernels/ |
| D | gru_cell.h | 27 void GruCell(const RuntimeShape& input_shape, const float* input,
28 const RuntimeShape& state_shape, const float* input_state,
29 const RuntimeShape& gate_weight_shape, const float* gate_weight,
30 const RuntimeShape& gate_bias_shape, const float* gate_bias,
31 const RuntimeShape& candidate_weight_shape,
33 const RuntimeShape& candidate_bias_shape,
34 const float* candidate_bias, const RuntimeShape& output_shape,
36 const RuntimeShape& activation_shape, float* activation,
37 const RuntimeShape& concat_shape, float* concat,
|
| D | gru_cell.cc | 32 void GruCell(const RuntimeShape& input_shape, const float* input, in GruCell()
33 const RuntimeShape& state_shape, const float* input_state, in GruCell()
34 const RuntimeShape& gate_weight_shape, const float* gate_weight, in GruCell()
35 const RuntimeShape& gate_bias_shape, const float* gate_bias, in GruCell()
36 const RuntimeShape& candidate_weight_shape, in GruCell()
38 const RuntimeShape& candidate_bias_shape, in GruCell()
39 const float* candidate_bias, const RuntimeShape& output_shape, in GruCell()
41 const RuntimeShape& activation_shape, float* activation, in GruCell()
42 const RuntimeShape& concat_shape, float* concat, in GruCell()
51 std::vector<RuntimeShape const*> concat_arrays_shapes; in GruCell()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/sparse_ops/ |
| D | fully_connected.h | 32 const RuntimeShape& input_shape, const float* input_data, in FullyConnectedSparseWeight()
33 const RuntimeShape& weights_shape, const float* weights_data, in FullyConnectedSparseWeight()
34 const RuntimeShape& bias_shape, const float* bias_data, in FullyConnectedSparseWeight()
35 const RuntimeShape& output_shape, float* output_data) { in FullyConnectedSparseWeight()
78 const RuntimeShape& input_shape, const float* input_data, in FullyConnectedSparseWeight1x4Impl()
79 const RuntimeShape& weights_shape, const float* weights_data, in FullyConnectedSparseWeight1x4Impl()
80 const RuntimeShape& bias_shape, const float* bias_data, in FullyConnectedSparseWeight1x4Impl()
81 const RuntimeShape& output_shape, float* output_data, int thread_start, in FullyConnectedSparseWeight1x4Impl()
118 const RuntimeShape& input_shape, const float* input_data, in FullyConnectedSparseWeight1x4Task()
119 const RuntimeShape& weights_shape, const float* weights_data, in FullyConnectedSparseWeight1x4Task()
[all …]
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ceva/ |
| D | types.h | 316 class RuntimeShape {
322 RuntimeShape& operator=(RuntimeShape const&) = delete;
324 RuntimeShape() : size_(0) {}
326 explicit RuntimeShape(int dimensions_count) : size_(dimensions_count) {
336 RuntimeShape(int shape_size, int32_t value) : size_(0) {
343 RuntimeShape(int dimensions_count, const int32_t* dims_data) : size_(0) {
347 RuntimeShape(const std::initializer_list<int> init_list) : size_(0) {
353 RuntimeShape(RuntimeShape const& other) : size_(other.DimensionsCount()) {
360 bool operator==(const RuntimeShape& comp) const {
365 ~RuntimeShape() {
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | depthwiseconv_multithread.h | 36 const RuntimeShape& input_shape, const T* input_data, in DepthwiseConvWorkerTask()
37 const RuntimeShape& filter_shape, in DepthwiseConvWorkerTask()
38 const T* filter_data, const RuntimeShape& bias_shape, in DepthwiseConvWorkerTask()
39 const TS* bias_data, const RuntimeShape& output_shape, in DepthwiseConvWorkerTask()
65 const RuntimeShape& input_shape_;
67 const RuntimeShape& filter_shape_;
69 const RuntimeShape& bias_shape_;
71 const RuntimeShape& output_shape_;
79 inline int HowManyConvThreads(const RuntimeShape& output_shape, in HowManyConvThreads()
80 const RuntimeShape& filter_shape) { in HowManyConvThreads()
[all …]
|
| D | optimized_ops.h | 132 VectorMap<Scalar> MapAsVector(Scalar* data, const RuntimeShape& shape) { in MapAsVector()
149 const RuntimeShape& shape) { in MapAsMatrixWithLastDimAsRows()
158 const RuntimeShape& shape) { in MapAsMatrixWithFirstDimAsCols()
173 const RuntimeShape& shape) { in MapAsArrayWithLastDimAsRows()
196 const RuntimeShape& shape, in MapAsMatrixWithGivenNumberOfRows()
206 const RuntimeShape& unswitched_input1_shape, in BinaryBroadcastFiveFold()
208 const RuntimeShape& unswitched_input2_shape, in BinaryBroadcastFiveFold()
210 const RuntimeShape& output_shape, in BinaryBroadcastFiveFold()
323 const RuntimeShape& bias_shape, in AddBiasAndEvalActivationFunction()
325 const RuntimeShape& array_shape, in AddBiasAndEvalActivationFunction()
[all …]
|
| D | batch_matmul.h | 28 inline void BatchMatMul(const RuntimeShape& lhs_shape, const float* lhs_data, in BatchMatMul()
29 const RuntimeShape& rhs_shape, const float* rhs_data, in BatchMatMul()
30 const RuntimeShape& output_shape, float* output_data, in BatchMatMul()
35 const RuntimeShape extended_lhs_shape = in BatchMatMul()
36 RuntimeShape::ExtendedShape(5, lhs_shape); in BatchMatMul()
37 const RuntimeShape extended_rhs_shape = in BatchMatMul()
38 RuntimeShape::ExtendedShape(5, rhs_shape); in BatchMatMul()
50 auto extent = [](const RuntimeShape& shape, int x) { in BatchMatMul()
115 inline void BatchMatMul(const RuntimeShape& lhs_shape, const int8_t* lhs_data, in BatchMatMul()
116 const RuntimeShape& rhs_shape, const int8_t* rhs_data, in BatchMatMul()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/ |
| D | depthwise_conv_hybrid.h | 42 const float* input_scales, const RuntimeShape& input_shape, in DepthwiseConvHybridGeneral()
43 const int8* input_data, const RuntimeShape& filter_shape, in DepthwiseConvHybridGeneral()
44 const int8* filter_data, const RuntimeShape& bias_shape, in DepthwiseConvHybridGeneral()
45 const float* bias_data, const RuntimeShape& output_shape, in DepthwiseConvHybridGeneral()
261 const RuntimeShape& input_shape, const int8* input_data, in DepthwiseConvHybridWithRounding()
262 const RuntimeShape& filter_shape, const int8* filter_data, in DepthwiseConvHybridWithRounding()
263 const RuntimeShape& bias_shape, const float* bias_data, in DepthwiseConvHybridWithRounding()
264 const RuntimeShape& output_shape, float* output_data, in DepthwiseConvHybridWithRounding()
320 const RuntimeShape& input_shape, const int8* input_data, in DepthwiseConvHybridImpl()
321 const RuntimeShape& filter_shape, const int8* filter_data, in DepthwiseConvHybridImpl()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/ |
| D | mul.h | 46 const RuntimeShape& input1_shape, const T* input1_data, in Mul()
47 const RuntimeShape& input2_shape, const T* input2_data, in Mul()
48 const RuntimeShape& output_shape, T* output_data) { in Mul()
60 const RuntimeShape& input1_shape, const int16_t* input1_data, in Mul()
61 const RuntimeShape& input2_shape, const int16_t* input2_data, in Mul()
62 const RuntimeShape& output_shape, int8_t* output_data) { in Mul()
90 const ArithmeticParams& params, const RuntimeShape& input1_shape, in BroadcastMul4DSlow()
91 const T* input1_data, const RuntimeShape& input2_shape, in BroadcastMul4DSlow()
92 const T* input2_data, const RuntimeShape& output_shape, T* output_data) { in BroadcastMul4DSlow()
100 const RuntimeShape extended_output_shape = in BroadcastMul4DSlow()
[all …]
|
| D | add.h | 49 const ArithmeticParams& params, const RuntimeShape& input1_shape, in BroadcastBinaryFunction4DSlow()
50 const int8_t* input1_data, const RuntimeShape& input2_shape, in BroadcastBinaryFunction4DSlow()
51 const int8_t* input2_data, const RuntimeShape& output_shape, in BroadcastBinaryFunction4DSlow()
59 const RuntimeShape extended_output_shape = in BroadcastBinaryFunction4DSlow()
60 RuntimeShape::ExtendedShape(4, output_shape); in BroadcastBinaryFunction4DSlow()
118 const RuntimeShape& input1_shape, const int8_t* input1_data, in Add()
119 const RuntimeShape& input2_shape, const int8_t* input2_data, in Add()
120 const RuntimeShape& output_shape, int8_t* output_data) { in Add()
130 const RuntimeShape& input1_shape, in BroadcastAdd4DSlow()
132 const RuntimeShape& input2_shape, in BroadcastAdd4DSlow()
[all …]
|