| /external/tensorflow/tensorflow/lite/kernels/ |
| D | maximum_minimum_test.cc | 28 MaxMinOpModel(tflite::BuiltinOperator op, const TensorData& input1, in MaxMinOpModel()
29 const TensorData& input2, const TensorType& output) { in MaxMinOpModel()
61 void TestModel(tflite::BuiltinOperator op, const TensorData& input1, in TestModel()
62 const TensorData& input2, const TensorData& output, in TestModel()
75 void TestModel(tflite::BuiltinOperator op, const TensorData& input1, in TestModel()
76 const TensorData& input2, const TensorData& output, in TestModel()
|
| D | detection_postprocess_test.cc | 40 BaseDetectionPostprocessOpModel(const TensorData& input1, in BaseDetectionPostprocessOpModel()
41 const TensorData& input2, in BaseDetectionPostprocessOpModel()
42 const TensorData& input3, in BaseDetectionPostprocessOpModel()
43 const TensorData& output1, in BaseDetectionPostprocessOpModel()
44 const TensorData& output2, in BaseDetectionPostprocessOpModel()
45 const TensorData& output3, in BaseDetectionPostprocessOpModel()
46 const TensorData& output4) { in BaseDetectionPostprocessOpModel()
252 const TensorData& input1, const TensorData& input2, in DetectionPostprocessOpModelwithRegularNMS()
253 const TensorData& input3, const TensorData& output1, in DetectionPostprocessOpModelwithRegularNMS()
254 const TensorData& output2, const TensorData& output3, in DetectionPostprocessOpModelwithRegularNMS()
[all …]
|
| D | reduce_test.cc | 60 MeanOpConstModel(const TensorData& input, const TensorData& output, in MeanOpConstModel()
75 MeanOpDynamicModel(const TensorData& input, const TensorData& output, in MeanOpDynamicModel()
76 const TensorData& axis, bool keep_dims) { in MeanOpDynamicModel()
89 SumOpConstModel(const TensorData& input, const TensorData& output, in SumOpConstModel()
104 SumOpDynamicModel(const TensorData& input, const TensorData& output, in SumOpDynamicModel()
105 const TensorData& axis, bool keep_dims) { in SumOpDynamicModel()
118 ProdOpConstModel(const TensorData& input, const TensorData& output, in ProdOpConstModel()
133 ProdOpDynamicModel(const TensorData& input, const TensorData& output, in ProdOpDynamicModel()
134 const TensorData& axis, bool keep_dims) { in ProdOpDynamicModel()
147 MaxOpConstModel(const TensorData& input, const TensorData& output, in MaxOpConstModel()
[all …]
|
| D | pad_test.cc | 72 PadV2OpConstModel(const TensorData& input, in PadV2OpConstModel()
75 const TensorData& output) { in PadV2OpConstModel()
89 PadV2OpConstModel(const TensorData& input, in PadV2OpConstModel()
92 const TensorData& constant_values, in PadV2OpConstModel()
93 const TensorData& output) { in PadV2OpConstModel()
115 PadOpConstModel(const TensorData& input, in PadOpConstModel()
118 const TensorData& output) { in PadOpConstModel()
134 PadV2OpDynamicModel(const TensorData& input, in PadV2OpDynamicModel()
137 const TensorData& output) { in PadV2OpDynamicModel()
148 PadV2OpDynamicModel(const TensorData& input, in PadV2OpDynamicModel()
[all …]
|
| D | test_util.h | 87 struct TensorData { struct
88 TensorData(TensorType type = TensorType_FLOAT32, std::vector<int> shape = {},
159 return AddInput(TensorData{type}, is_variable);
161 int AddInput(const TensorData& t, bool is_variable = false);
167 int id = AddTensor(TensorData{type, shape}, data); in AddConstInput()
176 int AddOutput(TensorType type) { return AddOutput(TensorData{type}); } in AddOutput()
177 int AddOutput(const TensorData& t);
362 int AddTensorPerChannelQuant(TensorData t) { in AddTensorPerChannelQuant()
381 int AddTensor(TensorData t, std::initializer_list<T> data,
469 std::map<int, TensorData> tensor_data_;
|
| D | mfcc_test.cc | 40 BaseMfccOpModel(const TensorData& input1, const TensorData& input2, in BaseMfccOpModel()
41 const TensorData& output) { in BaseMfccOpModel()
|
| D | floor_div_test.cc | 29 FloorDivModel(const TensorData& input1, const TensorData& input2, in FloorDivModel()
30 const TensorData& output) { in FloorDivModel()
|
| D | pow_test.cc | 30 PowOpModel(const TensorData& input1, const TensorData& input2, in PowOpModel()
31 const TensorData& output) { in PowOpModel()
|
| D | depthwise_conv_test.cc | 44 const TensorData& input, in BaseDepthwiseConvolutionOpModel()
45 const TensorData& filter, in BaseDepthwiseConvolutionOpModel()
46 const TensorData& output, in BaseDepthwiseConvolutionOpModel()
71 TensorData bias{TensorType_INT32, in BaseDepthwiseConvolutionOpModel()
85 TensorData bias{TensorType_INT32, {bias_size}, 0, 0, bias_scale}; in BaseDepthwiseConvolutionOpModel()
|
| D | floor_mod_test.cc | 29 FloorModModel(const TensorData& input1, const TensorData& input2, in FloorModModel()
30 const TensorData& output) { in FloorModModel()
|
| D | squared_difference_test.cc | 28 BaseSquaredDifferenceOpModel(const TensorData& input1, in BaseSquaredDifferenceOpModel()
29 const TensorData& input2, in BaseSquaredDifferenceOpModel()
30 const TensorData& output) { in BaseSquaredDifferenceOpModel()
|
| D | transpose_conv_test.cc | 41 const TensorData& filter, const TensorData& input, in TransposeConvOpModel()
42 const TensorData& output, Padding padding, int stride_w, in TransposeConvOpModel()
|
| D | space_to_batch_nd_test.cc | 71 SpaceToBatchNDOpConstModel(const TensorData& input, in SpaceToBatchNDOpConstModel()
74 const TensorData& output) { in SpaceToBatchNDOpConstModel()
97 SpaceToBatchNDOpDynamicModel(const TensorData& input, in SpaceToBatchNDOpDynamicModel()
98 const TensorData& output) { in SpaceToBatchNDOpDynamicModel()
|
| D | activations_test.cc | 31 BaseActivationsOpModel(BuiltinOperator type, TensorData input) { in BaseActivationsOpModel()
45 BaseActivationsOpModel(float softmax_beta, TensorData input) { in BaseActivationsOpModel()
59 BaseActivationsOpModel(BuiltinOperator type, const TensorData& input, in BaseActivationsOpModel()
60 const TensorData& output) { in BaseActivationsOpModel()
839 BasePReluOpModel(const TensorData& input, const TensorData& alpha) { in BasePReluOpModel()
943 LeakyReluOpModel(const TensorData& input, float alpha) { in LeakyReluOpModel()
|
| D | fully_connected_test.cc | 138 const TensorData& input, const TensorData& output = {TensorType_FLOAT32}, in BaseFullyConnectedOpModel()
163 TensorData bias{TensorType_INT32, {units_}, 0, 0, bias_scale};
276 HybridFullyConnectedOpModel(int units, int batches, const TensorData& input, in HybridFullyConnectedOpModel()
277 const TensorData& weights, in HybridFullyConnectedOpModel()
278 const TensorData& output = {TensorType_FLOAT32}) in HybridFullyConnectedOpModel()
289 TensorData bias{TensorType_FLOAT32, {units_}};
|
| D | neg_test.cc | 28 NegOpModel(const TensorData& input, const TensorData& output) { in NegOpModel()
|
| D | div_test.cc | 28 BaseDivOpModel(const TensorData& input1, const TensorData& input2, in BaseDivOpModel()
29 const TensorData& output, in BaseDivOpModel()
|
| D | mirror_pad_test.cc | 29 BaseMirrorPadOpModel(const TensorData& input, in BaseMirrorPadOpModel()
30 const TensorData& padding_matrix, in BaseMirrorPadOpModel()
31 const TensorData& output, in BaseMirrorPadOpModel()
|
| D | add_n_test.cc | 30 BaseAddNOpModel(const std::vector<TensorData>& inputs, in BaseAddNOpModel()
31 const TensorData& output) { in BaseAddNOpModel()
|
| D | audio_spectrogram_test.cc | 40 BaseAudioSpectrogramOpModel(const TensorData& input1, in BaseAudioSpectrogramOpModel()
41 const TensorData& output, int window_size, in BaseAudioSpectrogramOpModel()
|
| D | squeeze_test.cc | 29 BaseSqueezeOpModel(const TensorData& input, const TensorData& output, in BaseSqueezeOpModel()
|
| D | fill_test.cc | 29 explicit FillOpModel(const TensorData& input1, const TensorData& input2) { in FillOpModel()
|
| D | concatenation_test.cc | 32 BaseConcatenationOpModel(const TensorData& input_template, int axis, in BaseConcatenationOpModel()
64 QuantizedConcatenationOpModel(const std::vector<TensorData>& input_template, in QuantizedConcatenationOpModel()
66 const TensorData& output_template) { in QuantizedConcatenationOpModel()
|
| D | where_test.cc | 30 BaseWhereOpModel(const TensorData& input, const TensorData& output) { in BaseWhereOpModel()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 91 FloatAddOpModel(const TensorData& input1, const TensorData& input2, in FloatAddOpModel()
92 const TensorData& output, in FloatAddOpModel()
163 FloatMulOpModel(const TensorData& input1, const TensorData& input2, in FloatMulOpModel()
164 const TensorData& output, in FloatMulOpModel()
198 FloatPoolingOpModel(BuiltinOperator type, const TensorData& input, in FloatPoolingOpModel()
200 const TensorData& output) { in FloatPoolingOpModel()
266 const TensorData& input, const TensorData& filter, in ConvolutionOpModel()
267 const TensorData& output, int stride_width = 2, int stride_height = 2, in ConvolutionOpModel()
283 TensorData bias{TensorType_INT32, {bias_size}, 0, 0, bias_scale}; in ConvolutionOpModel()
449 DepthwiseConvolutionOpModel(const TensorData& input, const TensorData& filter, in DepthwiseConvolutionOpModel()
[all …]
|