Home
last modified time | relevance | path

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

123456

/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
Darithmetic_test.cc27 ArithmeticOpBaseModel(const TensorData& input1, const TensorData& input2, in ArithmeticOpBaseModel()
28 const TensorData& output) in ArithmeticOpBaseModel()
34 ArithmeticOpBaseModel(const TensorData& input1, const TensorData& input2, in ArithmeticOpBaseModel()
35 const TensorData& output, in ArithmeticOpBaseModel()
79 AddOpModel(const TensorData& input1, const TensorData& input2, in AddOpModel()
80 const TensorData& output, ActivationFunctionType activation_func) in AddOpModel()
83 AddOpModel(const TensorData& input1, const TensorData& input2, in AddOpModel()
84 const TensorData& output, in AddOpModel()
Dmin_max_builder_test.cc27 MinMaxOpModel(tflite::BuiltinOperator op, const TensorData& input1, in MinMaxOpModel()
28 const TensorData& input2, const TensorData& output) { in MinMaxOpModel()
37 MinMaxOpModel(tflite::BuiltinOperator op, const TensorData& input1, in MinMaxOpModel()
39 const TensorData& input2, in MinMaxOpModel()
41 const TensorData& output, bool input1_const) { in MinMaxOpModel()
82 void TestModel(tflite::BuiltinOperator op, const TensorData& input1, in TestModel()
83 const TensorData& input2, const TensorData& output, in TestModel()
100 void TestModelConstInput(tflite::BuiltinOperator op, const TensorData& input1, in TestModelConstInput()
101 const TensorData& input2, const TensorData& output, in TestModelConstInput()
/external/tensorflow/tensorflow/lite/kernels/
Datan2_custom_test.cc42 Atan2Model(tflite::TensorData y, tflite::TensorData x, in Atan2Model()
43 tflite::TensorData output) { in Atan2Model()
76 tflite::TensorData y = {GetTTEnum<Float>(), {}}; in TYPED_TEST()
77 tflite::TensorData x = {GetTTEnum<Float>(), {}}; in TYPED_TEST()
78 tflite::TensorData output = {GetTTEnum<Float>(), {}}; in TYPED_TEST()
91 tflite::TensorData y = {GetTTEnum<Float>(), {4, 2, 1}}; in TYPED_TEST()
92 tflite::TensorData x = {GetTTEnum<Float>(), {4, 2, 1}}; in TYPED_TEST()
93 tflite::TensorData output = {GetTTEnum<Float>(), {4, 2, 1}}; in TYPED_TEST()
Datan2_test.cc39 Atan2Model(tflite::TensorData y, in Atan2Model()
40 tflite::TensorData x, in Atan2Model()
41 tflite::TensorData output) { in Atan2Model()
77 tflite::TensorData y = {GetTTEnum<Float>(), {}}; in TYPED_TEST()
78 tflite::TensorData x = {GetTTEnum<Float>(), {}}; in TYPED_TEST()
79 tflite::TensorData output = {GetTTEnum<Float>(), {}}; in TYPED_TEST()
92 tflite::TensorData y = {GetTTEnum<Float>(), {4, 2, 1}}; in TYPED_TEST()
93 tflite::TensorData x = {GetTTEnum<Float>(), {4, 2, 1}}; in TYPED_TEST()
94 tflite::TensorData output = {GetTTEnum<Float>(), {4, 2, 1}}; in TYPED_TEST()
Dunsorted_segment_test.h35 UnsortedSegmentModel(const TensorData& data, const TensorData& segment_ids, in UnsortedSegmentModel()
36 const TensorData& num_segments, const BuiltinOperator op, in UnsortedSegmentModel()
48 const TensorData& data, const std::initializer_list<int>& segment_id_data, in UnsortedSegmentModel()
79 UnsortedSegmentModel<int32_t> getModel(const TensorData& data, in getModel()
80 const TensorData& segment_ids, in getModel()
81 const TensorData& num_segments) { in getModel()
86 const TensorData& data, const std::initializer_list<int>& segment_id_data, in getConstModel()
Dsign_test.cc39 SignModel(tflite::TensorData x, in SignModel()
40 tflite::TensorData output) { in SignModel()
70 tflite::TensorData x = {GetTTEnum<Float>(), {}}; in TYPED_TEST()
71 tflite::TensorData output = {GetTTEnum<Float>(), {}}; in TYPED_TEST()
83 tflite::TensorData x = {GetTTEnum<Float>(), {4, 2, 1}}; in TYPED_TEST()
84 tflite::TensorData output = {GetTTEnum<Float>(), {4, 2, 1}}; in TYPED_TEST()
Dsign_custom_test.cc42 SignModel(tflite::TensorData x, in SignModel()
43 tflite::TensorData output) { in SignModel()
73 tflite::TensorData x = {GetTTEnum<Float>(), {}}; in TYPED_TEST()
74 tflite::TensorData output = {GetTTEnum<Float>(), {}}; in TYPED_TEST()
86 tflite::TensorData x = {GetTTEnum<Float>(), {4, 2, 1}}; in TYPED_TEST()
87 tflite::TensorData output = {GetTTEnum<Float>(), {4, 2, 1}}; in TYPED_TEST()
Dreduce_test_common.h49 void SymmetricInt16Scaling(TensorData& tensor) { in SymmetricInt16Scaling()
74 BaseConstOpModel(TensorData input, TensorData output, in BaseConstOpModel()
94 BaseDynamicOpModel(TensorData input, TensorData output, in BaseDynamicOpModel()
95 const TensorData& axis, bool keep_dims) { in BaseDynamicOpModel()
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()
Dfully_connected_test.cc139 const TensorData& input, const TensorData& output = {TensorType_FLOAT32}, in BaseFullyConnectedOpModel()
217 TensorData bias{bias_type, {units_}, 0, 0, 0, 0, true,
222 TensorData bias{bias_type, {units_}, 0, 0, bias_scale};
284 const TensorData& input, const TensorData& output = {TensorType_INT8}, in QuantizedFullyConnectedOpModel()
359 const TensorData& input, in PerChannelQuantizedFullyConnectedOpModel()
361 const TensorData& output = {TensorType_INT8}, in PerChannelQuantizedFullyConnectedOpModel()
404 HybridFullyConnectedOpModel(int units, int batches, const TensorData& input, in HybridFullyConnectedOpModel()
405 const TensorData& weights, in HybridFullyConnectedOpModel()
406 const TensorData& output = {TensorType_FLOAT32}, in HybridFullyConnectedOpModel()
419 TensorData bias{TensorType_FLOAT32, {units_}};
[all …]
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()
45 const TensorData& input2, in MaxMinOpModel()
74 void TestModel(tflite::BuiltinOperator op, const TensorData& input1, in TestModel()
75 const TensorData& input2, const TensorData& output, in TestModel()
Dfloor_mod_test_common.h24 FloorModModel(const TensorData& input1, const TensorData& input2, in FloorModModel()
25 const TensorData& output) { in FloorModModel()
Dconv3d_transpose_test.cc38 std::initializer_list<int> output_shape_data, const TensorData& filter, in Conv3dTransposeOpModel()
39 const TensorData& input, const TensorData& bias, const TensorData& output, in Conv3dTransposeOpModel()
70 std::initializer_list<int> output_shape_data, const TensorData& filter, in Conv3dTransposeOpModel()
71 const TensorData& input, const TensorData& output, TestType test_type, in Conv3dTransposeOpModel()
Ddensify_test.cc44 DensifyOpModel(const TensorData& input, const std::vector<T>& input_data, in DensifyOpModel()
71 TensorData input = {}; in TEST()
85 TensorData input = {}; in TEST()
99 TensorData input = {}; in TEST()
Ddetection_postprocess_test.cc42 const TensorData& input1, const TensorData& input2, in BaseDetectionPostprocessOpModel()
43 const TensorData& input3, const TensorData& output1, in BaseDetectionPostprocessOpModel()
44 const TensorData& output2, const TensorData& output3, in BaseDetectionPostprocessOpModel()
45 const TensorData& output4, int max_classes_per_detection = 1) { in BaseDetectionPostprocessOpModel()
379 const TensorData& input1, const TensorData& input2, in DetectionPostprocessOpModelwithRegularNMS()
380 const TensorData& input3, const TensorData& output1, in DetectionPostprocessOpModelwithRegularNMS()
381 const TensorData& output2, const TensorData& output3, in DetectionPostprocessOpModelwithRegularNMS()
382 const TensorData& output4, bool use_regular_nms, int num_threads = 1) { in DetectionPostprocessOpModelwithRegularNMS()
606 TensorData input1, input2, input3; in TEST_P()
/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/
Dunsorted_segment_test.h35 UnsortedSegmentModel(const TensorData& data, const TensorData& segment_ids, in UnsortedSegmentModel()
36 const TensorData& num_segments, const BuiltinOperator op, in UnsortedSegmentModel()
48 const TensorData& data, const std::initializer_list<int>& segment_id_data, in UnsortedSegmentModel()
79 UnsortedSegmentModel<int32_t> getModel(const TensorData& data, in getModel()
80 const TensorData& segment_ids, in getModel()
81 const TensorData& num_segments) { in getModel()
86 const TensorData& data, const std::initializer_list<int>& segment_id_data, in getConstModel()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_nnapi_failure_handling_test.cc56 AddSubOpsAcceleratedModel(const TensorData& input1, const TensorData& input2, in AddSubOpsAcceleratedModel()
57 const TensorData& input3, const TensorData& output, in AddSubOpsAcceleratedModel()
88 void Init(const TensorData& input1, const TensorData& input2, in Init()
89 const TensorData& input3, const TensorData& output, in Init()
Dnnapi_delegate_c_api_test.cc69 FloatAddOpModel(const TensorData& input1, const TensorData& input2, in FloatAddOpModel()
70 const TensorData& output, in FloatAddOpModel()
76 const TensorData& input1, const TensorData& input2, in FloatAddOpModel()
77 const TensorData& output, in FloatAddOpModel()
95 void Init(const TensorData& input1, const TensorData& input2, in Init()
96 const TensorData& output, ActivationFunctionType activation_type) { in Init()
Dnnapi_delegate_errno_test.cc55 FloatAddOpModel(const NnApi* nnapi, const TensorData& input1, in FloatAddOpModel()
56 const TensorData& input2, const TensorData& output, in FloatAddOpModel()
75 void Init(const TensorData& input1, const TensorData& input2, in Init()
76 const TensorData& output, ActivationFunctionType activation_type, in Init()
Dnnapi_delegate_device_selection_test.cc46 const TensorData& input1, const TensorData& input2, in Init()
47 const TensorData& output, ActivationFunctionType activation_type, in Init()
375 AddSubOpsAcceleratedModel(const TensorData& input1, const TensorData& input2, in AddSubOpsAcceleratedModel()
376 const TensorData& input3, const TensorData& output, in AddSubOpsAcceleratedModel()
401 void Init(const TensorData& input1, const TensorData& input2, in Init()
402 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.cc167 FloatAddOpModel(const TensorData& input1, const TensorData& input2, in FloatAddOpModel()
168 const TensorData& output, in FloatAddOpModel()
175 const TensorData& input1, const TensorData& input2, in FloatAddOpModel()
176 const TensorData& output, in FloatAddOpModel()
195 void Init(const TensorData& input1, const TensorData& input2, in Init()
196 const TensorData& output, ActivationFunctionType activation_type, in Init()
534 FloatMulOpModel(const TensorData& input1, const TensorData& input2, in FloatMulOpModel()
535 const TensorData& output, in FloatMulOpModel()
568 FloatPoolingOpModel(BuiltinOperator type, const TensorData& input, in FloatPoolingOpModel()
570 const TensorData& output) { in FloatPoolingOpModel()
[all …]
/external/tensorflow/tensorflow/lite/kernels/gradient/
Dbcast_grad_args_test.cc35 BcastGradArgsInt32OpModel(const TensorData& input1, const TensorData& input2, in BcastGradArgsInt32OpModel()
36 const TensorData& output1, in BcastGradArgsInt32OpModel()
37 const TensorData& output2) { in BcastGradArgsInt32OpModel()
137 BcastGradArgsInt64OpModel(const TensorData& input1, const TensorData& input2, in BcastGradArgsInt64OpModel()
138 const TensorData& output1, in BcastGradArgsInt64OpModel()
139 const TensorData& output2) { in BcastGradArgsInt64OpModel()
/external/tensorflow/tensorflow/lite/delegates/coreml/
Dcoreml_executor.h23 struct TensorData { struct
36 - (bool)invokeWithInputs:(const std::vector<TensorData>&)inputs argument
37 outputs:(const std::vector<TensorData>&)outputs API_AVAILABLE(ios(11));

123456