| /external/libtextclassifier/utils/tflite/ |
| token_encoder.cc | 17 #include "utils/tflite/token_encoder.h"
21 #include "utils/tflite/encoder_common.h"
105 if (tflite::IsConstantTensor(&output_length)) {
108 tflite::SetTensorToDynamic(&output_positions);
112 tflite::SetTensorToDynamic(&output_attr);
129 if (!tflite::IsConstantTensor(&output_length)) {
177 namespace tflite { namespace
190 } // namespace tflite
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| lsh_projection.cc | 68 namespace tflite { namespace
203 } // namespace tflite
|
| /external/tensorflow/tensorflow/lite/ |
| arena_planner.cc | 18 namespace tflite { namespace
311 } // namespace tflite
|
| graph_info.cc | 19 namespace tflite { namespace
23 // Provide a range iterable wrapper for TfLiteIntArray* (C lists that TfLite
233 } // namespace tflite
|
| interpreter.cc | 35 namespace tflite { namespace
63 // TODO(b/128420794): Include the TFLite runtime version in the log.
282 } // namespace tflite
|
| interpreter.h | 15 // Main abstraction controlling the tflite interpreter.
33 namespace tflite { namespace
503 // The error reporter delegate that tflite will forward queries errors to.
521 } // namespace tflite
|
| /external/tensorflow/tensorflow/lite/experimental/kernels/ |
| ctc_beam_entry.h | 29 namespace tflite { namespace
148 } // namespace tflite
|
| top_n.h | 52 namespace tflite { namespace
339 } // namespace tflite
|
| /external/tensorflow/tensorflow/lite/experimental/micro/kernels/ |
| depthwise_conv_test.cc | 23 namespace tflite { namespace
61 ::tflite::ops::micro::AllOpsResolver resolver;
63 resolver.FindOp(tflite::BuiltinOperator_DEPTHWISE_CONV_2D, 1);
141 ::tflite::ops::micro::AllOpsResolver resolver;
143 resolver.FindOp(tflite::BuiltinOperator_DEPTHWISE_CONV_2D, 1);
191 } // namespace tflite
198 tflite::testing::TestDepthwiseConvFloat( //
225 using tflite::testing::F2Q;
226 using tflite::testing::F2Q32;
239 tflite::testing::TestDepthwiseConvQuantized( /
[all...] |
| fully_connected.cc | 24 namespace tflite { namespace
90 tflite::FullyConnectedParams op_params;
131 tflite::FullyConnectedParams op_params;
134 tflite::reference_ops::FullyConnected(
184 } // namespace tflite
|
| fully_connected_test.cc | 23 namespace tflite { namespace
56 ::tflite::ops::micro::AllOpsResolver resolver;
58 resolver.FindOp(tflite::BuiltinOperator_FULLY_CONNECTED, 1);
134 ::tflite::ops::micro::AllOpsResolver resolver;
136 resolver.FindOp(tflite::BuiltinOperator_FULLY_CONNECTED, 1);
182 } // namespace tflite
189 tflite::testing::TestFullyConnectedFloat( //
215 tflite::testing::TestFullyConnectedFloat( //
239 tflite::testing::TestFullyConnectedFloat( //
263 using tflite::testing::F2Q
[all...] |
| softmax.cc | 25 namespace tflite { namespace
49 tflite::PreprocessSoftmaxScaling(
52 data->diff_min = -1.0 * tflite::CalculateInputRadius(
74 tflite::reference_ops::Softmax(input->data.f, input_size, 1, params->beta,
83 tflite::reference_ops::Softmax(input->data.f, input_size, batch_size,
100 tflite::reference_ops::Softmax(op_params, shape,
119 tflite::reference_ops::Softmax(op_params, shape,
129 tflite::reference_ops::Softmax(
140 tflite::reference_ops::Softmax(
213 } // namespace tflite
[all...] |
| softmax_test.cc | 23 namespace tflite { namespace
47 ::tflite::ops::micro::AllOpsResolver resolver;
49 resolver.FindOp(tflite::BuiltinOperator_SOFTMAX, 1);
113 ::tflite::ops::micro::AllOpsResolver resolver;
115 resolver.FindOp(tflite::BuiltinOperator_SOFTMAX, 1);
158 } // namespace tflite
165 tflite::testing::TestSoftmaxFloat( //
189 using tflite::testing::F2Q;
197 tflite::testing::TestSoftmaxQuantized( //
|
| /external/tensorflow/tensorflow/lite/experimental/microfrontend/ |
| audio_microfrontend.cc | 22 namespace tflite { namespace
206 } // namespace tflite
|
| audio_microfrontend_test.cc | 15 // Unit test for TFLite Micro Frontend op.
27 namespace tflite { namespace
200 } // namespace tflite
203 ::tflite::LogToStderr();
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| add_test.cc | 21 namespace tflite { namespace
421 } // namespace tflite
423 ::tflite::LogToStderr();
|
| audio_spectrogram.cc | 27 namespace tflite { namespace
165 } // namespace tflite
|
| basic_rnn_test.cc | 15 // Unit test for TFLite RNN op.
29 namespace tflite { namespace
342 } // namespace tflite
345 ::tflite::LogToStderr();
|
| batch_to_space_nd.cc | 25 namespace tflite { namespace
204 } // namespace tflite
|
| conv_test.cc | 24 namespace tflite { namespace
[all...] |
| depthwise_conv_test.cc | 23 namespace tflite { namespace
545 } // namespace tflite
548 ::tflite::LogToStderr();
|
| detection_postprocess_test.cc | 26 namespace tflite { namespace
572 } // namespace tflite
575 ::tflite::LogToStderr();
|
| div.cc | 24 namespace tflite { namespace
85 tflite::ArithmeticParams op_params; \
179 } // namespace tflite
|
| elementwise.cc | 22 namespace tflite { namespace
179 } // namespace tflite
|
| embedding_lookup_sparse.cc | 74 namespace tflite { namespace
250 } // namespace tflite
|
