Home
last modified time | relevance | path

Searched refs:inputs_size (Results 1 – 25 of 60) sorted by relevance

123

/external/tensorflow/tensorflow/lite/micro/kernels/
Dactivations_test.cc34 constexpr int inputs_size = 1; in TestReluFloat() local
36 constexpr int tensors_size = inputs_size + outputs_size; in TestReluFloat()
67 constexpr int inputs_size = 1; in TestRelu6Float() local
69 constexpr int tensors_size = inputs_size + outputs_size; in TestRelu6Float()
103 constexpr int inputs_size = 1; in TestReluUint8() local
105 constexpr int tensors_size = inputs_size + outputs_size; in TestReluUint8()
144 constexpr int inputs_size = 1; in TestRelu6Uint8() local
146 constexpr int tensors_size = inputs_size + outputs_size; in TestRelu6Uint8()
184 constexpr int inputs_size = 1; in TestReluInt8() local
186 constexpr int tensors_size = inputs_size + outputs_size; in TestReluInt8()
[all …]
Dzeros_like_test.cc32 constexpr int inputs_size = 1; in TestZerosLikeFloat() local
34 constexpr int tensors_size = inputs_size + outputs_size; in TestZerosLikeFloat()
64 constexpr int inputs_size = 1; in TestZerosLikeInt32() local
66 constexpr int tensors_size = inputs_size + outputs_size; in TestZerosLikeInt32()
96 constexpr int inputs_size = 1; in TestZerosLikeInt64() local
98 constexpr int tensors_size = inputs_size + outputs_size; in TestZerosLikeInt64()
Dcast_test.cc33 constexpr int inputs_size = 1; in TestCastFloatToInt8() local
35 constexpr int tensors_size = inputs_size + outputs_size; in TestCastFloatToInt8()
65 constexpr int inputs_size = 1; in TestCastInt8ToFloat() local
67 constexpr int tensors_size = inputs_size + outputs_size; in TestCastInt8ToFloat()
Dtranspose_conv_test.cc115 constexpr int inputs_size = 4; in TestTransposeConvFloat() local
117 constexpr int tensors_size = inputs_size + outputs_size; in TestTransposeConvFloat()
160 constexpr int inputs_size = 4; in TestTransposeConvQuantized() local
162 constexpr int tensors_size = inputs_size + outputs_size; in TestTransposeConvQuantized()
234 constexpr int inputs_size = 4; in TF_LITE_MICRO_TEST() local
236 constexpr int tensors_size = inputs_size + outputs_size; in TF_LITE_MICRO_TEST()
269 constexpr int inputs_size = 4; in TF_LITE_MICRO_TEST() local
271 constexpr int tensors_size = inputs_size + outputs_size; in TF_LITE_MICRO_TEST()
Dmaximum_minimum_test.cc37 constexpr int inputs_size = 2; in TestMaxMinFloat() local
39 constexpr int tensors_size = inputs_size + outputs_size; in TestMaxMinFloat()
78 constexpr int inputs_size = 2; in TestMaxMinQuantized() local
80 constexpr int tensors_size = inputs_size + outputs_size; in TestMaxMinQuantized()
117 constexpr int inputs_size = 2; in TestMaxMinQuantizedInt32() local
119 constexpr int tensors_size = inputs_size + outputs_size; in TestMaxMinQuantizedInt32()
Dpad_test.cc101 constexpr int inputs_size = 2; in TestPadFloat() local
103 constexpr int tensors_size = inputs_size + outputs_size; in TestPadFloat()
128 constexpr int inputs_size = 3; in TestPadV2Float() local
130 constexpr int tensors_size = inputs_size + outputs_size; in TestPadV2Float()
156 constexpr int inputs_size = 2; in TestPadQuantized() local
158 constexpr int tensors_size = inputs_size + outputs_size; in TestPadQuantized()
195 constexpr int inputs_size = 3; in TestPadV2Quantized() local
197 constexpr int tensors_size = inputs_size + outputs_size; in TestPadV2Quantized()
Dconv_test_common.cc127 constexpr int inputs_size = 3; in TestConvFloat() local
129 constexpr int tensors_size = inputs_size + outputs_size; in TestConvFloat()
160 constexpr int inputs_size = 3; in TestConvQuantizedPerLayer() local
162 constexpr int tensors_size = inputs_size + outputs_size; in TestConvQuantizedPerLayer()
229 constexpr int inputs_size = 3; in TestConvQuantizedPerChannel() local
231 constexpr int tensors_size = inputs_size + outputs_size; in TestConvQuantizedPerChannel()
Dcircular_buffer_test.cc59 constexpr int inputs_size = 2; in TF_LITE_MICRO_TEST() local
61 constexpr int tensors_size = inputs_size + outputs_size; in TF_LITE_MICRO_TEST()
127 constexpr int inputs_size = 2; in TF_LITE_MICRO_TEST() local
129 constexpr int tensors_size = inputs_size + outputs_size; in TF_LITE_MICRO_TEST()
Dtanh_test.cc76 constexpr int inputs_size = 1; in TestTanhFloat() local
78 constexpr int tensors_size = inputs_size + outputs_size; in TestTanhFloat()
118 constexpr int inputs_size = 1; in TestTanhQuantized() local
120 constexpr int tensors_size = inputs_size + outputs_size; in TestTanhQuantized()
Dprelu_test.cc56 constexpr int inputs_size = 2; in TestPreluFloat() local
58 constexpr int tensors_size = inputs_size + outputs_size; in TestPreluFloat()
84 constexpr int inputs_size = 2; in TestPreluQuantized() local
86 constexpr int tensors_size = inputs_size + outputs_size; in TestPreluQuantized()
Dlogistic_test.cc78 constexpr int inputs_size = 1; in TestLogisticFloat() local
80 constexpr int tensors_size = inputs_size + outputs_size; in TestLogisticFloat()
101 constexpr int inputs_size = 1; in TestLogisticQuantized() local
103 constexpr int tensors_size = inputs_size + outputs_size; in TestLogisticQuantized()
Dbatch_to_space_nd_test.cc76 constexpr int inputs_size = 3; in TestBatchToSpaceNdFloat() local
78 constexpr int tensors_size = inputs_size + outputs_size; in TestBatchToSpaceNdFloat()
103 constexpr int inputs_size = 3; in TestBatchToSpaceNdQuantized() local
105 constexpr int tensors_size = inputs_size + outputs_size; in TestBatchToSpaceNdQuantized()
Dspace_to_batch_nd_test.cc76 constexpr int inputs_size = 3; in TestSpaceToBatchNdFloat() local
78 constexpr int tensors_size = inputs_size + outputs_size; in TestSpaceToBatchNdFloat()
103 constexpr int inputs_size = 3; in TestSpaceToBatchNdQuantized() local
105 constexpr int tensors_size = inputs_size + outputs_size; in TestSpaceToBatchNdQuantized()
Dhard_swish_test.cc93 constexpr int inputs_size = 1; in TestHardSwishQuantized() local
95 constexpr int tensors_size = inputs_size + outputs_size; in TestHardSwishQuantized()
172 constexpr int inputs_size = 1; in TestHardSwishQuantizedBias() local
174 constexpr int tensors_size = inputs_size + outputs_size; in TestHardSwishQuantizedBias()
226 constexpr int inputs_size = 1; in TestHardSwishFloat() local
228 constexpr int tensors_size = inputs_size + outputs_size; in TestHardSwishFloat()
Dround_test.cc32 constexpr int inputs_size = 1; in TestRound() local
34 constexpr int tensors_size = inputs_size + outputs_size; in TestRound()
Dceil_test.cc32 constexpr int inputs_size = 1; in TestCeil() local
34 constexpr int tensors_size = inputs_size + outputs_size; in TestCeil()
Dexp_test.cc33 constexpr int inputs_size = 1; in TestExp() local
35 constexpr int tensors_size = inputs_size + outputs_size; in TestExp()
Dfloor_test.cc33 constexpr int inputs_size = 1; in TestFloor() local
35 constexpr int tensors_size = inputs_size + outputs_size; in TestFloor()
Dneg_test.cc33 constexpr int inputs_size = 1; in TestNegFloat() local
35 constexpr int tensors_size = inputs_size + outputs_size; in TestNegFloat()
Darg_min_max_test.cc59 constexpr int inputs_size = 2; in TestArgMinMaxFloat() local
61 constexpr int tensors_size = inputs_size + outputs_size; in TestArgMinMaxFloat()
85 constexpr int inputs_size = 2; in TestArgMinMaxQuantized() local
87 constexpr int tensors_size = inputs_size + outputs_size; in TestArgMinMaxQuantized()
Dpooling_test.cc72 constexpr int inputs_size = 1; in TestAveragePoolFloat() local
74 constexpr int tensors_size = inputs_size + outputs_size; in TestAveragePoolFloat()
103 constexpr int inputs_size = 1; in TestAveragePoolQuantized() local
105 constexpr int tensors_size = inputs_size + outputs_size; in TestAveragePoolQuantized()
130 constexpr int inputs_size = 1; in TestMaxPoolFloat() local
132 constexpr int tensors_size = inputs_size + outputs_size; in TestMaxPoolFloat()
159 constexpr int inputs_size = 1; in TestMaxPoolQuantized() local
161 constexpr int tensors_size = inputs_size + outputs_size; in TestMaxPoolQuantized()
Dmul_test.cc79 constexpr int inputs_size = 2; in TestMulFloat() local
81 constexpr int tensors_size = inputs_size + outputs_size; in TestMulFloat()
106 constexpr int inputs_size = 2; in TestMulQuantized() local
108 constexpr int tensors_size = inputs_size + outputs_size; in TestMulQuantized()
Ddepthwise_conv_test.cc98 constexpr int inputs_size = 3; in TestDepthwiseConvFloat() local
100 constexpr int tensors_size = inputs_size + outputs_size; in TestDepthwiseConvFloat()
127 constexpr int inputs_size = 3; in TestDepthwiseConvQuantizedPerLayer() local
129 constexpr int tensors_size = inputs_size + outputs_size; in TestDepthwiseConvQuantizedPerLayer()
213 constexpr int inputs_size = 3; in TestDepthwiseConvQuantizedPerChannel() local
215 constexpr int tensors_size = inputs_size + outputs_size; in TestDepthwiseConvQuantizedPerChannel()
752 constexpr int inputs_size = 3; in TF_LITE_MICRO_TEST() local
754 constexpr int tensors_size = inputs_size + outputs_size; in TF_LITE_MICRO_TEST()
833 constexpr int inputs_size = 3; in TF_LITE_MICRO_TEST() local
835 constexpr int tensors_size = inputs_size + outputs_size; in TF_LITE_MICRO_TEST()
/external/tensorflow/tensorflow/lite/python/optimize/
Dcalibration_wrapper.cc174 const size_t inputs_size = PyList_Size(input_shapes); in Prepare() local
175 if (inputs_size != interpreter_->inputs().size()) { in Prepare()
178 interpreter_->inputs().size(), inputs_size); in Prepare()
182 for (size_t i = 0; i < inputs_size; i++) { in Prepare()
212 const size_t inputs_size = PyList_Size(input_value); in FeedTensor() local
214 if (inputs_size != interpreter_->inputs().size()) { in FeedTensor()
217 interpreter_->inputs().size(), inputs_size); in FeedTensor()
221 for (size_t i = 0; i < inputs_size; i++) { in FeedTensor()
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dgradients_test.py72 def lstm_model_fn(batch_size, state_size, steps, inputs_size=None): argument
73 inputs_size = inputs_size or state_size
75 random_ops.random_normal([batch_size, inputs_size]) for _ in range(steps)
111 def create_lstm_batch_jacobian(batch_size, state_size, steps, inputs_size=None): argument
113 inputs_size=inputs_size)
185 def create_lstm_per_eg_grad(batch_size, state_size, steps, inputs_size=None): argument
186 inputs_size = inputs_size or state_size
188 random_ops.random_normal([batch_size, inputs_size]) for _ in range(steps)
495 inputs_size=128)
605 100, 32, 8, inputs_size=128)

123