Home
last modified time | relevance | path

Searched refs:TensorData (Results 1 – 25 of 118) sorted by relevance

12345

/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
Darithmetic_test.cc25 ArithmeticOpBaseModel(const TensorData& input1, const TensorData& input2, in ArithmeticOpBaseModel()
26 const TensorData& output) in ArithmeticOpBaseModel()
32 ArithmeticOpBaseModel(const TensorData& input1, const TensorData& input2, in ArithmeticOpBaseModel()
33 const TensorData& output, in ArithmeticOpBaseModel()
77 AddOpModel(const TensorData& input1, const TensorData& input2, in AddOpModel()
78 const TensorData& output, ActivationFunctionType activation_func) in AddOpModel()
81 AddOpModel(const TensorData& input1, const TensorData& input2, in AddOpModel()
82 const TensorData& output, in AddOpModel()
Dmin_max_builder_test.cc25 MinMaxOpModel(tflite::BuiltinOperator op, const TensorData& input1, in MinMaxOpModel()
26 const TensorData& input2, const TensorData& output) { in MinMaxOpModel()
35 MinMaxOpModel(tflite::BuiltinOperator op, const TensorData& input1, in MinMaxOpModel()
37 const TensorData& input2, in MinMaxOpModel()
39 const TensorData& output, bool input1_const) { in MinMaxOpModel()
80 void TestModel(tflite::BuiltinOperator op, const TensorData& input1, in TestModel()
81 const TensorData& input2, const TensorData& output, in TestModel()
98 void TestModelConstInput(tflite::BuiltinOperator op, const TensorData& input1, in TestModelConstInput()
99 const TensorData& input2, const TensorData& output, in TestModelConstInput()
Dstrided_slice_test.cc24 StridedSliceOpModel(const TensorData& input, const TensorData& output, in StridedSliceOpModel()
25 const TensorData& begin, in StridedSliceOpModel()
27 const TensorData& end, in StridedSliceOpModel()
29 const TensorData& strides, in StridedSliceOpModel()
Dtranspose_conv_test.cc26 const TensorData& filter, in QuantizedTransposeConvOpModel()
28 const TensorData& input, in QuantizedTransposeConvOpModel()
29 const TensorData& output, Padding padding, in QuantizedTransposeConvOpModel()
53 TensorData bias{TensorType_INT32, in QuantizedTransposeConvOpModel()
67 TensorData bias{TensorType_INT32, {bias_size}, 0, 0, bias_scale}; in QuantizedTransposeConvOpModel()
Dsquared_difference_test.cc23 SquaredDifferenceOpModel(const TensorData& input1, const TensorData& input2, in SquaredDifferenceOpModel()
24 const TensorData& output) { in SquaredDifferenceOpModel()
Dslice_test.cc24 SliceOpModel(const TensorData& input, const TensorData& output, in SliceOpModel()
25 const TensorData& begin, const TensorData& size, in SliceOpModel()
Dmul_test.cc24 explicit MulOpModel(const TensorData& input1, const TensorData& input2, in MulOpModel()
25 const TensorData& output, in MulOpModel()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_nnapi_failure_handling_test.cc55 AddSubOpsAcceleratedModel(const TensorData& input1, const TensorData& input2, in AddSubOpsAcceleratedModel()
56 const TensorData& input3, const TensorData& output, in AddSubOpsAcceleratedModel()
86 void Init(const TensorData& input1, const TensorData& input2, in Init()
87 const TensorData& input3, const TensorData& output, in Init()
Dnnapi_delegate_errno_test.cc52 FloatAddOpModel(const NnApi* nnapi, const TensorData& input1, in FloatAddOpModel()
53 const TensorData& input2, const TensorData& output, in FloatAddOpModel()
72 void Init(const TensorData& input1, const TensorData& input2, in Init()
73 const TensorData& output, ActivationFunctionType activation_type, in Init()
Dnnapi_delegate_device_selection_test.cc45 const TensorData& input1, const TensorData& input2, in Init()
46 const TensorData& output, ActivationFunctionType activation_type, in Init()
374 AddSubOpsAcceleratedModel(const TensorData& input1, const TensorData& input2, in AddSubOpsAcceleratedModel()
375 const TensorData& input3, const TensorData& output, in AddSubOpsAcceleratedModel()
400 void Init(const TensorData& input1, const TensorData& input2, in Init()
401 const TensorData& input3, const TensorData& output, in Init()
561 HardSwishAddOpsAcceleratedModel(const TensorData& input1, in HardSwishAddOpsAcceleratedModel()
562 const TensorData& input2, in HardSwishAddOpsAcceleratedModel()
563 const TensorData& output, in HardSwishAddOpsAcceleratedModel()
585 void Init(const TensorData& input1, const TensorData& input2, in Init()
[all …]
Dnnapi_delegate_test.cc154 FloatAddOpModel(const TensorData& input1, const TensorData& input2, in FloatAddOpModel()
155 const TensorData& output, in FloatAddOpModel()
162 const TensorData& input1, const TensorData& input2, in FloatAddOpModel()
163 const TensorData& output, in FloatAddOpModel()
182 void Init(const TensorData& input1, const TensorData& input2, in Init()
183 const TensorData& output, ActivationFunctionType activation_type, in Init()
490 FloatMulOpModel(const TensorData& input1, const TensorData& input2, in FloatMulOpModel()
491 const TensorData& output, in FloatMulOpModel()
524 FloatPoolingOpModel(BuiltinOperator type, const TensorData& input, in FloatPoolingOpModel()
526 const TensorData& output) { in FloatPoolingOpModel()
[all …]
Dnnapi_delegate_signed_quantization_test.cc65 const TensorData& input, in HybridFullyConnectedOpModel()
66 const TensorData& weights, in HybridFullyConnectedOpModel()
67 const TensorData& output = {TensorType_FLOAT32}, in HybridFullyConnectedOpModel()
80 TensorData bias{TensorType_FLOAT32, {units_}};
230 const TensorData& input, const TensorData& filter, in BaseConvolutionOpModel()
231 const TensorData& output, int stride_width = 2, int stride_height = 2, in BaseConvolutionOpModel()
270 TensorData bias{bias_type,
284 TensorData bias{TensorType_INT32, {bias_size}, 0, 0, bias_scale};
691 QuantizeOpModel(const NnApi* nnapi, const TensorData& input, in QuantizeOpModel()
692 const TensorData& output) { in QuantizeOpModel()
[all …]
/external/tensorflow/tensorflow/lite/kernels/
Dcomplex_support_test.cc32 RealOpModel(const TensorData& input, const TensorData& output) { in RealOpModel()
93 ImagOpModel(const TensorData& input, const TensorData& output) { in ImagOpModel()
154 ComplexAbsOpModel(const TensorData& input, const TensorData& output) { in ComplexAbsOpModel()
Dconv3d_test.cc32 Conv3dOpModel(const TensorData& input, const TensorData& filter, in Conv3dOpModel()
33 const TensorData& bias, const TensorData& output, in Conv3dOpModel()
52 Conv3dOpModel(const TensorData& input, const TensorData& filter, in Conv3dOpModel()
53 const TensorData& output, Padding padding = Padding_VALID, in Conv3dOpModel()
Ddepthwise_conv_hybrid_test.cc54 TfLiteRegistration* registration, const TensorData& input, in BaseDepthwiseConvolutionOpModel()
55 const TensorData& filter, const TensorData& output, Padding padding_type, in BaseDepthwiseConvolutionOpModel()
81 TensorData bias{TensorType_INT32, in BaseDepthwiseConvolutionOpModel()
95 TensorData bias{TensorType_INT32, {bias_size}, 0, 0, bias_scale}; in BaseDepthwiseConvolutionOpModel()
196 const TensorData input({TensorType_FLOAT32, {b, h, w, c}}); in RandomTest()
197 const TensorData output({TensorType_FLOAT32, {}}); in RandomTest()
204 const TensorData filter({TensorType_INT8, in RandomTest()
Dreduce_test.cc56 TensorData& SymmetricInt16Scaling(TensorData& tensor) { in SymmetricInt16Scaling()
82 MeanOpConstModel(TensorData input, TensorData output, in MeanOpConstModel()
97 MeanOpDynamicModel(const TensorData& input, const TensorData& output, in MeanOpDynamicModel()
98 const TensorData& axis, bool keep_dims) { in MeanOpDynamicModel()
111 SumOpConstModel(const TensorData& input, const TensorData& output, in SumOpConstModel()
126 SumOpDynamicModel(const TensorData& input, const TensorData& output, in SumOpDynamicModel()
127 const TensorData& axis, bool keep_dims) { in SumOpDynamicModel()
140 ProdOpConstModel(const TensorData& input, const TensorData& output, in ProdOpConstModel()
155 ProdOpDynamicModel(const TensorData& input, const TensorData& output, in ProdOpDynamicModel()
156 const TensorData& axis, bool keep_dims) { in ProdOpDynamicModel()
[all …]
Ddensify_test.cc44 DensifyOpModel(const TensorData& input, const std::vector<T>& input_data, in DensifyOpModel()
69 TensorData input = {}; in TEST()
83 TensorData input = {}; in TEST()
97 TensorData input = {}; in TEST()
Dmaximum_minimum_test.cc34 MaxMinOpModel(tflite::BuiltinOperator op, const TensorData& input1, in MaxMinOpModel()
35 const TensorData& input2, const TensorType& output) { in MaxMinOpModel()
44 MaxMinOpModel(tflite::BuiltinOperator op, const TensorData& input1, in MaxMinOpModel()
46 const TensorData& input2, in MaxMinOpModel()
75 void TestModel(tflite::BuiltinOperator op, const TensorData& input1, in TestModel()
76 const TensorData& input2, const TensorData& output, in TestModel()
Ddetection_postprocess_test.cc41 BaseDetectionPostprocessOpModel(const TensorData& input1, in BaseDetectionPostprocessOpModel()
42 const TensorData& input2, in BaseDetectionPostprocessOpModel()
43 const TensorData& input3, in BaseDetectionPostprocessOpModel()
44 const TensorData& output1, in BaseDetectionPostprocessOpModel()
45 const TensorData& output2, in BaseDetectionPostprocessOpModel()
46 const TensorData& output3, in BaseDetectionPostprocessOpModel()
47 const TensorData& output4) { in BaseDetectionPostprocessOpModel()
317 const TensorData& input1, const TensorData& input2, in DetectionPostprocessOpModelwithRegularNMS()
318 const TensorData& input3, const TensorData& output1, in DetectionPostprocessOpModelwithRegularNMS()
319 const TensorData& output2, const TensorData& output3, in DetectionPostprocessOpModelwithRegularNMS()
[all …]
Dfully_connected_test.cc139 const TensorData& input, const TensorData& output = {TensorType_FLOAT32}, in BaseFullyConnectedOpModel()
170 TensorData bias{TensorType_INT64, {units_}, 0, 0, bias_scale};
173 TensorData bias{TensorType_INT32, {units_}, 0, 0, bias_scale};
296 HybridFullyConnectedOpModel(int units, int batches, const TensorData& input, in HybridFullyConnectedOpModel()
297 const TensorData& weights, in HybridFullyConnectedOpModel()
298 const TensorData& output = {TensorType_FLOAT32}, in HybridFullyConnectedOpModel()
311 TensorData bias{TensorType_FLOAT32, {units_}};
1186 int batches, const TensorData& input, in SparseFullyConnectedOpModel()
1187 const TensorData& weights, in SparseFullyConnectedOpModel()
1202 TensorData bias{input.type, {units_}}; in SparseFullyConnectedOpModel()
[all …]
Dpad_test.cc76 PadV2OpConstModel(const TensorData& input, in PadV2OpConstModel()
79 const TensorData& output) { in PadV2OpConstModel()
93 PadV2OpConstModel(const TensorData& input, in PadV2OpConstModel()
96 const TensorData& constant_values, in PadV2OpConstModel()
97 const TensorData& output) { in PadV2OpConstModel()
119 PadOpConstModel(const TensorData& input, in PadOpConstModel()
122 const TensorData& output) { in PadOpConstModel()
138 PadV2OpDynamicModel(const TensorData& input, in PadV2OpDynamicModel()
141 const TensorData& output) { in PadV2OpDynamicModel()
152 PadV2OpDynamicModel(const TensorData& input, in PadV2OpDynamicModel()
[all …]
Dmfcc_test.cc38 BaseMfccOpModel(const TensorData& input1, const TensorData& input2, in BaseMfccOpModel()
39 const TensorData& output) { in BaseMfccOpModel()
/external/tensorflow/tensorflow/lite/delegates/coreml/
Dcoreml_executor.h24 struct TensorData { struct
37 - (bool)invokeWithInputs:(const std::vector<TensorData>&)inputs argument
38 outputs:(const std::vector<TensorData>&)outputs API_AVAILABLE(ios(11));
/external/tensorflow/tensorflow/lite/kernels/perception/
Ddense_image_warp_test.cc34 DenseImageWarpOpModel(const TensorData& input, const TensorData& flow, in DenseImageWarpOpModel()
35 const TensorData& output) { in DenseImageWarpOpModel()
/external/tensorflow/tensorflow/lite/kernels/parse_example/
Dparse_example_test.cc164 string_indices_.push_back(AddInput(TensorData(TensorType_STRING, {1}))); in ParseExampleOpModel()
167 AddConstInput<std::string>(TensorData(TensorType_STRING, {0}), {""})); in ParseExampleOpModel()
169 string_indices_.push_back(AddInput(TensorData(TensorType_STRING, {1}))); in ParseExampleOpModel()
172 string_indices_.push_back(AddInput(TensorData(TensorType_STRING, {1}))); in ParseExampleOpModel()
176 TensorData(dense_types[0], {dense_size}), dense_defaults); in ParseExampleOpModel()

12345