Home
last modified time | relevance | path

Searched defs:input_dims_data (Results 1 – 25 of 38) sorted by relevance

12

/external/tensorflow/tensorflow/lite/micro/kernels/
Dzeros_like_test.cc27 void TestZerosLikeFloat(const int* input_dims_data, const float* input_data, in TestZerosLikeFloat()
58 void TestZerosLikeInt32(const int* input_dims_data, const int32_t* input_data, in TestZerosLikeInt32()
90 void TestZerosLikeInt64(const int* input_dims_data, const int64_t* input_data, in TestZerosLikeInt64()
Dactivations_test.cc27 void TestReluFloat(const int* input_dims_data, const float* input_data, in TestReluFloat()
60 void TestRelu6Float(const int* input_dims_data, const float* input_data, in TestRelu6Float()
93 void TestReluUint8(const int* input_dims_data, const float* input_data, in TestReluUint8()
134 void TestRelu6Uint8(const int* input_dims_data, const float* input_data, in TestRelu6Uint8()
175 void TestReluInt8(const int* input_dims_data, const float* input_data, in TestReluInt8()
215 void TestRelu6Int8(const int* input_dims_data, const float* input_data, in TestRelu6Int8()
Dcast_test.cc27 void TestCastFloatToInt8(const int* input_dims_data, const float* input_data, in TestCastFloatToInt8()
59 void TestCastInt8ToFloat(const int* input_dims_data, const int8_t* input_data, in TestCastInt8ToFloat()
Ddequantize_test.cc50 void TestDequantizeToFloat(const int* input_dims_data, const float* input_data, in TestDequantizeToFloat()
72 void TestDequantizeToInt32(const int* input_dims_data, const float* input_data, in TestDequantizeToInt32()
Dlogistic_test.cc71 void TestLogisticFloat(const int* input_dims_data, const float* input_data, in TestLogisticFloat()
91 void TestLogisticQuantized(const int* input_dims_data, const float* input_data, in TestLogisticQuantized()
Dbatch_to_space_nd_test.cc67 const int* input_dims_data, const float* input_data, in TestBatchToSpaceNdFloat()
92 const int* input_dims_data, const float* input_data, T* input_quantized, in TestBatchToSpaceNdQuantized()
Dspace_to_batch_nd_test.cc67 const int* input_dims_data, const float* input_data, in TestSpaceToBatchNdFloat()
92 const int* input_dims_data, const float* input_data, T* input_quantized, in TestSpaceToBatchNdQuantized()
Dtanh_test.cc68 void TestTanhFloat(const int input_dims_data[], const float* input_data, in TestTanhFloat()
103 void TestTanhQuantized(const int input_dims_data[], const float* input_data, in TestTanhQuantized()
Dunpack_test.cc27 const int* input_dims_data, const float* input_data, int axis, in TestUnpackThreeOutputsFloat()
93 void TestUnpackOneOutputFloat(const int* input_dims_data, in TestUnpackOneOutputFloat()
137 const int* input_dims_data, const uint8_t* input_data, int axis, in TestUnpackThreeOutputsQuantized()
207 const int* input_dims_data, const int32_t* input_data, int axis, in TestUnpackThreeOutputsQuantized32()
Dreshape_test.cc115 void TestReshape(const int* input_dims_data, const float* input_data, in TestReshape()
134 void TestReshapeQuantized(const int* input_dims_data, const T* input_data, in TestReshapeQuantized()
212 int input_dims_data[] = {3, 1, 2, 2}; in TF_LITE_MICRO_TEST() local
332 int input_dims_data[] = {1, 1}; in TF_LITE_MICRO_TEST() local
Dexp_test.cc28 void TestExp(const int* input_dims_data, const float* input_data, in TestExp()
Dhard_swish_test.cc64 const int input_dims_data[] = {2, 1, size}; in TestHardSwishQuantized() local
163 const int input_dims_data[] = {2, 1, size}; in TestHardSwishQuantizedBias() local
217 const int input_dims_data[] = {1, size}; in TestHardSwishFloat() local
Dprelu_test.cc48 void TestPreluFloat(const int* input_dims_data, const float* input_data, in TestPreluFloat()
72 void TestPreluQuantized(const int* input_dims_data, const float* input_data, in TestPreluQuantized()
Delementwise_test.cc27 const int* input_dims_data, const float* input_data, in TestElementwiseFloat()
64 const int* input_dims_data, const bool* input_data, in TestElementwiseBool()
Dround_test.cc27 void TestRound(const int* input_dims_data, const float* input_data, in TestRound()
Dceil_test.cc27 void TestCeil(const int* input_dims_data, const float* input_data, in TestCeil()
Dfloor_test.cc27 void TestFloor(const int* input_dims_data, const float* input_data, in TestFloor()
Dneg_test.cc27 void TestNegFloat(const int* input_dims_data, const float* input_data, in TestNegFloat()
Dconv_test_common.cc114 TfLiteStatus TestConvFloat(const int* input_dims_data, const float* input_data, in TestConvFloat()
143 const int* input_dims_data, const float* input_data, in TestConvQuantizedPerLayer()
185 const int* input_dims_data, const float* input_data, in TestConvQuantizedPerChannel()
Dpad_test.cc92 void TestPadFloat(const int* input_dims_data, const float* input_data, in TestPadFloat()
117 void TestPadV2Float(const int* input_dims_data, const float* input_data, in TestPadV2Float()
145 void TestPadQuantized(const int* input_dims_data, const float* input_data, in TestPadQuantized()
178 void TestPadV2Quantized(const int* input_dims_data, const float* input_data, in TestPadV2Quantized()
Dsplit_test.cc28 const int* input_dims_data, const float* input_data, in TestSplitTwoOutputsFloat()
82 const int* input_dims_data, const float* input_data, in TestSplitFourOutputsFloat()
156 const int* input_dims_data, const uint8_t* input_data, in TestSplitTwoOutputsQuantized()
212 const int* input_dims_data, const int32_t* input_data, in TestSplitTwoOutputsQuantized32()
Dreduce_test.cc92 void TestMeanFloatInput4D(const int* input_dims_data, const float* input_data, in TestMeanFloatInput4D()
120 void TestReduceOpFloat(const int* input_dims_data, const float* input_data, in TestReduceOpFloat()
149 const int* input_dims_data, const float* input_data, T* input_data_quant, in TestReduceOpQuantized()
185 void TestMeanOpQuantized(const int* input_dims_data, const float* input_data, in TestMeanOpQuantized()
Dtranspose_conv_test.cc100 const int* input_dims_data, const float* input_data, in TestTransposeConvFloat()
132 const int* input_dims_data, const float* input_data, in TestTransposeConvQuantized()
Darg_min_max_test.cc50 void TestArgMinMaxFloat(const int* input_dims_data, const float* input_values, in TestArgMinMaxFloat()
73 void TestArgMinMaxQuantized(const int* input_dims_data, in TestArgMinMaxQuantized()
Dshape_test.cc47 void TestShape(const int* input_dims_data, const float* input_data, in TestShape()

12