Home
last modified time | relevance | path

Searched refs:SetBias (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dconv_test.cc140 void SetBias(std::initializer_list<float> f) { PopulateTensor(bias_, f); } in SetBias() function in tflite::__anone58ff41e0111::ConvolutionOpModel
183 m.SetBias({1, 2, 3}); in TEST_P()
214 m.SetBias({1, 2, 3}); in TEST_P()
249 m.SetBias({1, 2, 3}); in TEST_P()
278 m.SetBias({0}); in TEST_P()
307 m.SetBias({1, 2, 3}); in TEST_P()
341 m.SetBias({1, 2, 3}); in TEST_P()
375 m.SetBias({1, 2, 3}); in TEST_P()
407 m.SetBias({1, 2, 3}); in TEST_P()
440 m.SetBias({0}); in TEST_P()
[all …]
Ddepthwise_conv_test.cc139 void SetBias(std::initializer_list<float> f) { PopulateTensor(bias_, f); } in SetBias() function in tflite::__anon1154144a0111::DepthwiseConvolutionOpModel
183 m.SetBias({1, 2, 3, 4}); in TEST_P()
214 m.SetBias({1, 2, 3, 4}); in TEST_P()
245 m.SetBias({1, 2, 3, 4}); in TEST_P()
276 m.SetBias({1, 2, 3, 4}); in StrideTest()
314 m.SetBias({1, 2, 3, 4}); in PaddingTest()
349 m.SetBias({1, 2, 3, 4}); in SimpleTest()
410 m.SetBias({0}); in SimpleDilatedTestPaddingValid()
455 m.SetBias({0}); in SimpleDilatedTestPaddingSame()
513 m.SetBias({0, 0, 0, 0}); in BatchPaddingValidTest()
[all …]
Dfully_connected_test.cc218 void SetBias(const std::vector<float>& f) { PopulateTensor(bias_, f); } in SetBias() function in tflite::__anonc847e4c60111::FloatFullyConnectedOpModel
237 void SetBias(const std::vector<float>& data) { in SetBias() function in tflite::__anonc847e4c60111::QuantizedFullyConnectedOpModel
330 void SetBias(const std::vector<float>& f) { PopulateTensor(bias_, f); } in SetBias() function in tflite::__anonc847e4c60111::HybridFullyConnectedOpModel
408 m.SetBias({1, 2, 3}); in TEST_P()
427 m.SetBias({1}); in TEST_P()
475 m.SetBias({1, 2, 3}); in TEST_P()
539 m.SetBias({1, 2, 3}); in TEST_P()
625 m.SetBias({1, 2, 3, 4}); in TEST_P()
651 m.SetBias({1, 2, 3}); in TEST_P()
682 m.SetBias({1, 2, 3}); in TEST_P()
[all …]
Ddepthwise_conv_hybrid_test.cc140 void SetBias(std::initializer_list<float> data) { in SetBias() function in tflite::__anone347176b0111::PerChannelHybridDepthwiseConvolutionOpModel
152 void SetBias(const std::vector<float>& data) { PopulateTensor(bias_, data); } in SetBias() function in tflite::__anone347176b0111::PerChannelHybridDepthwiseConvolutionOpModel
222 hybrid_generic.SetBias(bias_data); in RandomTest()
233 hybrid_optimized.SetBias(bias_data); in RandomTest()
360 m.SetBias({3, -2, 4, 6}); in TEST_P()
399 m.SetBias({0, 0, 0, 0, 0, 0, 0, 0}); in TEST_P()
435 m.SetBias({0, 0, 0, 0, 0, 0, 0, 0}); in TEST_P()
Dunidirectional_sequence_rnn_test.cc209 void SetBias(std::initializer_list<float> f) { PopulateTensor(bias_, f); } in SetBias() function in tflite::__anona6a394470111::UnidirectionalRNNOpModel
283 rnn.SetBias(rnn_bias); in TEST()
312 rnn.SetBias(rnn_bias); in TEST_P()
339 rnn.SetBias(rnn_bias); in TEST_P()
365 rnn.SetBias(rnn_bias); in TEST()
395 rnn.SetBias(rnn_bias); in TEST_P()
426 rnn.SetBias(rnn_bias); in TEST_P()
Dbasic_rnn_test.cc195 void SetBias(std::initializer_list<float> f) { PopulateTensor(bias_, f); } in SetBias() function in tflite::__anonaa6237b70111::RNNOpModel
262 rnn.SetBias(rnn_bias); in TEST()
291 rnn.SetBias(rnn_bias); in TEST_P()
319 rnn.SetBias(rnn_bias); in TEST_P()
Dconv3d_test.cc73 void SetBias(std::initializer_list<float> f) { PopulateTensor(bias_, f); } in SetBias() function in tflite::__anon352068d50111::Conv3dOpModel
247 m.SetBias({1, 2}); in TEST()
Dtranspose_conv_test.cc649 void SetBias(std::initializer_list<float> bias) { in SetBias() function in tflite::__anonbe94865c0111::BaseTransposeConvBiasOpModel
698 model.SetBias({3, 4}); in TEST_P()
732 model.SetBias({1}); in TEST_P()
774 model.SetBias({1}); in TEST_P()
Dsvdf_test.cc509 void SetBias(const std::vector<float>& f) { in SetBias() function in tflite::__anon2a1e039b0111::IntegerSVDFOpModel
556 svdf.SetBias({-0.0976817, 0.15294972, 0.39635518, -0.02702999}); in TEST_F()
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
Dconv_test.cc111 void SetBias(std::initializer_list<float> data) { in SetBias() function in tflite::QuantizedConvolutionOpModel
161 m.SetBias({1, 2, 3}); in TEST()
195 m.SetBias({1, 2, 3}); in TEST()
232 m.SetBias({1, 2, 3}); in TEST()
386 m.SetBias({0}); in TEST()
417 m.SetBias({1, 2}); in TEST()
457 m.SetBias({1, 2, 3}); in TEST()
499 m.SetBias({1, 2, 3}); in TEST()
Dmatmul_test.cc71 void SetBias(const std::vector<float>& data) { in SetBias() function in tflite::FullyConnectedOpModel
122 m.SetBias({1, 2, 3}); in TEST_P()
150 m.SetBias({1, 2, 3}); in TEST_P()
232 m.SetBias({1, 2, 3}); in TEST_P()
260 m.SetBias({1, 2, 3}); in TEST_P()
296 m.SetBias({1, 2, 3}); in TEST()
Dtranspose_conv_test.cc86 void SetBias(std::initializer_list<float> bias) { in SetBias() function in tflite::QuantizedTransposeConvOpModel
241 model.SetBias({1}); in TEST()
262 model.SetBias({1}); in TEST()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_signed_quantization_test.cc101 void SetBias(const std::vector<float>& f) { PopulateTensor(bias_, f); } in SetBias() function in tflite::__anonb9fc30f60111::HybridFullyConnectedOpModel
166 m.SetBias({1, 2, 3}); in TEST_F()
202 m.SetBias({1, 2, 3}); in TEST_F()
325 void SetBias(std::initializer_list<float> data) { in SetBias() function in tflite::__anonb9fc30f60111::QuantizedConvolutionOpModel
356 m.SetBias({1, 2, 3}); in TEST_F()
396 m.SetBias({1, 2, 3}); in TEST_F()
428 void SetBias(std::initializer_list<float> data) { in SetBias() function in tflite::__anonb9fc30f60111::PerChannelQuantizedConvolutionOpModel
479 m.SetBias({3, -2}); in TEST_F()
540 m.SetBias({3, -2}); in TEST_F()
600 m.SetBias({3, -2}); in TEST_F()
[all …]
Dnnapi_delegate_test.cc633 void SetBias(std::initializer_list<float> data) { in SetBias() function in tflite::__anon0c68fcb70111::ConvolutionOpModel
685 m.SetBias({1, 2, 3}); in TEST()
723 m.SetBias({1, 2, 3}); in TEST()
755 m.SetBias({1, 2, 3}); in TEST()
792 quant_op.SetBias(bias); in TEST()
797 float_op.SetBias(bias); in TEST()
851 m.SetBias({0}); in TEST()
874 void SetBias(std::initializer_list<float> data) { in SetBias() function in tflite::__anon0c68fcb70111::QuantizedConvolutionOpModel
936 m.SetBias({0}); in TEST()
1101 void SetBias(std::initializer_list<float> data) { in SetBias() function in tflite::__anon0c68fcb70111::DepthwiseConvolutionOpModel
[all …]
Dnnapi_delegate_device_selection_test.cc742 void SetBias(std::initializer_list<float> data) { in SetBias() function in tflite::__anon2e62bd8b0111::QuantizedWeightsConvolutionOpModel
797 m.SetBias({1, 2, 3}); in TEST_F()
/external/tensorflow/tensorflow/lite/delegates/coreml/builders/
Dfully_connected_op_builder.cc36 void FullyConnectedOpBuilder::SetBias(TfLiteTensor* bias) { bias_ = bias; } in SetBias() function in tflite::delegates::coreml::FullyConnectedOpBuilder
114 SetBias(&context->tensors[inputs->data[kBias]]); in RegisterInputs()
Dfully_connected_op_builder.h42 void SetBias(TfLiteTensor* bias);
Dconvolution_op_builder.h48 void SetBias(TfLiteTensor* bias);
Dconvolution_op_builder.cc37 void ConvolutionOpBuilder::SetBias(TfLiteTensor* bias) { bias_ = bias; } in SetBias() function in tflite::delegates::coreml::ConvolutionOpBuilder
263 SetBias(&context->tensors[inputs->data[2]]); in RegisterInputs()