| /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()
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 | 129 class RuntimeShape {
135 RuntimeShape& operator=(RuntimeShape const&) = delete;
137 RuntimeShape() : size_(0) {} in RuntimeShape() function
139 explicit RuntimeShape(int dimensions_count) : size_(dimensions_count) { in RuntimeShape() function
149 RuntimeShape(int shape_size, int32 value) : size_(0) { in RuntimeShape() function
156 RuntimeShape(int dimensions_count, const int32* dims_data) : size_(0) { in RuntimeShape() function
160 RuntimeShape(const std::initializer_list<int> init_list) : size_(0) { in RuntimeShape() function
166 RuntimeShape(RuntimeShape const& other) : size_(other.DimensionsCount()) { in RuntimeShape() function
173 bool operator==(const RuntimeShape& comp) const {
178 ~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,
|
| D | depthwiseconv_quantized_test.cc | 121 const DepthwiseParams& params, const RuntimeShape& input_shape, in DispatchDepthwiseConvGeneral()
124 const RuntimeShape& filter_shape, in DispatchDepthwiseConvGeneral()
127 const RuntimeShape& bias_shape, const int32* bias_data, in DispatchDepthwiseConvGeneral()
130 const RuntimeShape& output_shape, in DispatchDepthwiseConvGeneral()
141 const DepthwiseParams& params, const RuntimeShape& input_shape, in DispatchDepthwiseConvGeneral()
142 const int8* input_data, const RuntimeShape& filter_shape, in DispatchDepthwiseConvGeneral()
143 const int8* filter_data, const RuntimeShape& bias_shape, in DispatchDepthwiseConvGeneral()
146 const RuntimeShape& output_shape, int8* output_data, int thread_start, in DispatchDepthwiseConvGeneral()
157 const RuntimeShape& input_shape, in DispatchDepthwiseConvImpl()
160 const RuntimeShape& filter_shape, in DispatchDepthwiseConvImpl()
[all …]
|
| D | depthwiseconv_per_channel_quantized_test.cc | 41 const DepthwiseParams& params, const RuntimeShape& input_shape, in PickOutputMultiplier()
42 const int8* input_data, const RuntimeShape& filter_shape, in PickOutputMultiplier()
43 const int8* filter_data, const RuntimeShape& bias_shape, in PickOutputMultiplier()
44 const int32* bias_data, const RuntimeShape& output_shape, in PickOutputMultiplier()
117 const RuntimeShape& input_shape_inference, const std::int8_t* input_data, in PickReasonableMultiplier()
118 const RuntimeShape& filter_shape_inference, const std::int8_t* filter_data, in PickReasonableMultiplier()
119 const RuntimeShape& bias_shape_inference, const std::int32_t* bias_data, in PickReasonableMultiplier()
120 const RuntimeShape& output_shape_inference, in PickReasonableMultiplier()
193 RuntimeShape* input_shape_inference, RuntimeShape* filter_shape_inference, in GenerateValidShapeConfigurations()
194 RuntimeShape* output_shape_inference, int* pad_width, int* pad_height, in GenerateValidShapeConfigurations()
[all …]
|
| D | depthwiseconv_float_test.cc | 34 const DepthwiseParams& params, const RuntimeShape& input_shape, in TestOneDepthwiseConv()
35 const float* input_data, const RuntimeShape& filter_shape, in TestOneDepthwiseConv()
36 const float* filter_data, const RuntimeShape& bias_shape, in TestOneDepthwiseConv()
37 const float* bias_data, const RuntimeShape& output_shape) { in TestOneDepthwiseConv()
103 RuntimeShape input_shape_inference( in TryTestOneDepthwiseConv()
105 RuntimeShape output_shape_inference; in TryTestOneDepthwiseConv()
115 RuntimeShape filter_shape_inference( in TryTestOneDepthwiseConv()
117 RuntimeShape bias_shape_inference({1, 1, 1, output_depth}); in TryTestOneDepthwiseConv()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | comparisons.h | 57 const ComparisonParams& op_params, const RuntimeShape& input1_shape, in ComparisonImpl()
58 const T* input1_data, const RuntimeShape& input2_shape, in ComparisonImpl()
59 const T* input2_data, const RuntimeShape& output_shape, bool* output_data) { in ComparisonImpl()
69 const RuntimeShape& input1_shape, in Comparison()
71 const RuntimeShape& input2_shape, in Comparison()
73 const RuntimeShape& output_shape, bool* output_data) { in Comparison()
80 const ComparisonParams& op_params, const RuntimeShape& input1_shape, in ComparisonWithScaling()
81 const T* input1_data, const RuntimeShape& input2_shape, in ComparisonWithScaling()
82 const T* input2_data, const RuntimeShape& output_shape, bool* output_data) { in ComparisonWithScaling()
111 const RuntimeShape& unextended_input1_shape, const T* input1_data, in BroadcastComparison4DSlowImpl()
[all …]
|
| D | reference_ops.h | 69 const RuntimeShape& unextended_input_shape, in DepthToSpace()
71 const RuntimeShape& unextended_output_shape, in DepthToSpace()
75 const RuntimeShape input_shape = in DepthToSpace()
76 RuntimeShape::ExtendedShape(4, unextended_input_shape); in DepthToSpace()
77 const RuntimeShape output_shape = in DepthToSpace()
78 RuntimeShape::ExtendedShape(4, unextended_output_shape); in DepthToSpace()
122 const RuntimeShape& unextended_input_shape, in SpaceToDepth()
124 const RuntimeShape& unextended_output_shape, in SpaceToDepth()
128 const RuntimeShape input_shape = in SpaceToDepth()
129 RuntimeShape::ExtendedShape(4, unextended_input_shape); in SpaceToDepth()
[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* pad_value_ptr, const RuntimeShape& output_shape, in Pad()
[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()
120 const RuntimeShape& input1_shape, const uint8* input1_data, in Add()
121 const RuntimeShape& input2_shape, const uint8* input2_data, in Add()
122 const RuntimeShape& output_shape, uint8* output_data) { in Add()
136 const RuntimeShape& input1_shape, const int16* input1_data, in Add()
[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* input1_data, in Mul()
64 const RuntimeShape& input2_shape, const uint8* input2_data, in Mul()
65 const RuntimeShape& output_shape, uint8* 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 | binary_function.h | 34 const RuntimeShape& unextended_input1_shape, const T1* input1_data, in BroadcastBinaryFunction4DSlow()
35 const RuntimeShape& unextended_input2_shape, const T2* input2_data, in BroadcastBinaryFunction4DSlow()
36 const RuntimeShape& unextended_output_shape, R* output_data, in BroadcastBinaryFunction4DSlow()
41 const RuntimeShape output_shape = in BroadcastBinaryFunction4DSlow()
42 RuntimeShape::ExtendedShape(4, unextended_output_shape); in BroadcastBinaryFunction4DSlow()
68 inline void BinaryFunction(const RuntimeShape& input1_shape, in BinaryFunction()
70 const RuntimeShape& input2_shape, in BinaryFunction()
72 const RuntimeShape& output_shape, R* output_data, in BinaryFunction()
|
| D | fully_connected.h | 29 const FullyConnectedParams& params, const RuntimeShape& input_shape, in FullyConnected()
30 const float* input_data, const RuntimeShape& weights_shape, in FullyConnected()
31 const float* weights_data, const RuntimeShape& bias_shape, in FullyConnected()
32 const float* bias_data, const RuntimeShape& output_shape, in FullyConnected()
65 const FullyConnectedParams& params, const RuntimeShape& input_shape, in FullyConnected()
66 const uint8* input_data, const RuntimeShape& filter_shape, in FullyConnected()
67 const uint8* filter_data, const RuntimeShape& bias_shape, in FullyConnected()
68 const int32* bias_data, const RuntimeShape& output_shape, in FullyConnected()
113 const FullyConnectedParams& params, const RuntimeShape& input_shape, in FullyConnected()
114 const uint8* input_data, const RuntimeShape& filter_shape, in FullyConnected()
[all …]
|
| D | logistic.h | 30 inline void Logistic(const RuntimeShape& input_shape, const float* input_data, in Logistic()
31 const RuntimeShape& output_shape, float* output_data) { in Logistic()
61 inline void Logistic(const LogisticParams&, const RuntimeShape& input_shape, in Logistic()
62 const float* input_data, const RuntimeShape& output_shape, in Logistic()
69 const RuntimeShape& input_shape, const int16* input_data, in Logistic()
70 const RuntimeShape& output_shape, int16* output_data) { in Logistic()
|
| D | conv.h | 28 inline void Conv(const ConvParams& params, const RuntimeShape& input_shape, in Conv()
29 const float* input_data, const RuntimeShape& filter_shape, in Conv()
30 const float* filter_data, const RuntimeShape& bias_shape, in Conv()
31 const float* bias_data, const RuntimeShape& output_shape, in Conv()
32 float* output_data, const RuntimeShape& im2col_shape, in Conv()
101 inline void Conv(const ConvParams& params, const RuntimeShape& input_shape, in Conv()
102 const uint8* input_data, const RuntimeShape& filter_shape, in Conv()
103 const uint8* filter_data, const RuntimeShape& bias_shape, in Conv()
104 const int32* bias_data, const RuntimeShape& output_shape, in Conv()
105 uint8* output_data, const RuntimeShape& im2col_shape, in Conv()
[all …]
|
| D | process_broadcast_shapes.h | 38 inline bool ProcessBroadcastShapes(const RuntimeShape& shape0, in ProcessBroadcastShapes()
39 const RuntimeShape& shape1, in ProcessBroadcastShapes()
45 RuntimeShape scalar_shape(dims_count, 1); in ProcessBroadcastShapes()
47 auto extended_shape0 = RuntimeShape::ExtendedShape(dims_count, shape0); in ProcessBroadcastShapes()
48 auto extended_shape1 = RuntimeShape::ExtendedShape(dims_count, shape1); in ProcessBroadcastShapes()
86 const RuntimeShape* shape_a = in ProcessBroadcastShapes()
88 const RuntimeShape* shape_b = in ProcessBroadcastShapes()
|
| D | strided_slice.h | 27 const RuntimeShape& unextended_input_shape, in StridedSlice()
29 const RuntimeShape& unextended_output_shape, in StridedSlice()
36 const RuntimeShape input_shape = in StridedSlice()
37 RuntimeShape::ExtendedShape(4, unextended_input_shape); in StridedSlice()
38 const RuntimeShape output_shape = in StridedSlice()
39 RuntimeShape::ExtendedShape(4, unextended_output_shape); in StridedSlice()
|
| /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/ |
| 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 | 126 VectorMap<Scalar> MapAsVector(Scalar* data, const RuntimeShape& shape) { in MapAsVector()
143 const RuntimeShape& shape) { in MapAsMatrixWithLastDimAsRows()
152 const RuntimeShape& shape) { in MapAsMatrixWithFirstDimAsCols()
167 const RuntimeShape& shape) { in MapAsArrayWithLastDimAsRows()
190 const RuntimeShape& shape, in MapAsMatrixWithGivenNumberOfRows()
265 const RuntimeShape& bias_shape, in AddBiasAndEvalActivationFunction()
267 const RuntimeShape& array_shape, in AddBiasAndEvalActivationFunction()
275 const FullyConnectedParams& params, const RuntimeShape& input_shape, in FullyConnected()
276 const float* input_data, const RuntimeShape& weights_shape, in FullyConnected()
277 const float* weights_data, const RuntimeShape& bias_shape, in FullyConnected()
[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 | 44 const RuntimeShape& input1_shape, const int8_t* input1_data, in Mul()
45 const RuntimeShape& input2_shape, const int8_t* input2_data, in Mul()
46 const RuntimeShape& output_shape, int8_t* output_data) { in Mul()
58 const RuntimeShape& input1_shape, const int16* input1_data, in Mul()
59 const RuntimeShape& input2_shape, const int16* input2_data, in Mul()
60 const RuntimeShape& output_shape, int8_t* output_data) { in Mul()
87 const RuntimeShape& input1_shape, in BroadcastMul4DSlow()
89 const RuntimeShape& input2_shape, in BroadcastMul4DSlow()
91 const RuntimeShape& output_shape, in BroadcastMul4DSlow()
99 const RuntimeShape extended_output_shape = in BroadcastMul4DSlow()
[all …]
|
| 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()
66 const FullyConnectedParams& params, const RuntimeShape& input_shape, in FullyConnected()
67 const int16_t* input_data, const RuntimeShape& filter_shape, in FullyConnected()
68 const int8_t* filter_data, const RuntimeShape& bias_shape, in FullyConnected()
69 const int64_t* bias_data, const RuntimeShape& output_shape, in FullyConnected()
|