Home
last modified time | relevance | path

Searched refs:tensors_size (Results 1 – 25 of 28) sorted by relevance

12

/external/tensorflow/tensorflow/lite/experimental/micro/kernels/
Dsoftmax_test.cc38 constexpr int tensors_size = inputs_size + outputs_size; in TestSoftmaxFloat() local
39 TfLiteTensor tensors[tensors_size] = { in TestSoftmaxFloat()
45 PopulateContext(tensors, tensors_size, &context); in TestSoftmaxFloat()
102 constexpr int tensors_size = inputs_size + outputs_size; in TestSoftmaxQuantized() local
103 TfLiteTensor tensors[tensors_size] = { in TestSoftmaxQuantized()
111 PopulateContext(tensors, tensors_size, &context); in TestSoftmaxQuantized()
Ddepthwise_conv_test.cc45 constexpr int tensors_size = inputs_size + outputs_size; in TestDepthwiseConvFloat() local
46 TfLiteTensor tensors[tensors_size] = { in TestDepthwiseConvFloat()
59 PopulateContext(tensors, tensors_size, &context); in TestDepthwiseConvFloat()
126 constexpr int tensors_size = inputs_size + outputs_size; in TestDepthwiseConvQuantized() local
127 TfLiteTensor tensors[tensors_size] = { in TestDepthwiseConvQuantized()
139 PopulateContext(tensors, tensors_size, &context); in TestDepthwiseConvQuantized()
Dfully_connected_test.cc45 constexpr int tensors_size = inputs_size + outputs_size; in TestFullyConnectedFloat() local
46 TfLiteTensor tensors[tensors_size] = { in TestFullyConnectedFloat()
54 PopulateContext(tensors, tensors_size, &context); in TestFullyConnectedFloat()
119 constexpr int tensors_size = inputs_size + outputs_size; in TestFullyConnectedQuantized() local
120 TfLiteTensor tensors[tensors_size] = { in TestFullyConnectedQuantized()
132 PopulateContext(tensors, tensors_size, &context); in TestFullyConnectedQuantized()
/external/tensorflow/tensorflow/lite/experimental/micro/
Dmicro_interpreter.h42 size_t tensors_size() const { return context_.tensors_size; } in tensors_size() function
Dmicro_interpreter_test.cc108 constexpr size_t tensors_size = 3; in BuildMockModel() local
109 const Offset<Tensor> tensors[tensors_size] = { in BuildMockModel()
138 CreateSubGraph(builder, builder.CreateVector(tensors, tensors_size), in BuildMockModel()
Dmicro_interpreter.cc82 context_.tensors_size = tensors_->Length(); in MicroInterpreter()
85 sizeof(TfLiteTensor) * context_.tensors_size, 4)); in MicroInterpreter()
/external/tensorflow/tensorflow/lite/experimental/micro/testing/
Dtest_utils.h81 inline void PopulateContext(TfLiteTensor* tensors, int tensors_size, in PopulateContext() argument
83 context->tensors_size = tensors_size; in PopulateContext()
/external/tensorflow/tensorflow/lite/
Doptional_debug_tools.cc82 interpreter->tensors_size(), interpreter->nodes_size()); in PrintInterpreterState()
88 for (size_t tensor_index = 0; tensor_index < interpreter->tensors_size(); in PrintInterpreterState()
Dinterpreter.cc244 TF_LITE_ENSURE(context_, tensor_index < tensors_size()); in SetBufferHandle()
264 TF_LITE_ENSURE(context_, tensor_index < tensors_size()); in GetBufferHandle()
Dinterpreter.h235 size_t tensors_size() const { return context_->tensors_size; } in tensors_size() function
Dmodel_test.cc130 ASSERT_EQ(interpreter->tensors_size(), 4); in TEST()
Dinterpreter_test.cc106 ASSERT_EQ(interpreter.tensors_size(), 0); in TEST()
108 ASSERT_EQ(interpreter.tensors_size(), 2); in TEST()
111 ASSERT_EQ(interpreter.tensors_size(), 5); in TEST()
113 ASSERT_EQ(interpreter.tensors_size(), 6); in TEST()
Dnnapi_delegate.cc121 for (size_t i = 0; i < subgraph->tensors_size(); i++) { in addTensorOperands()
711 std::vector<int64_t> tensor_id_to_nnapi_id(subgraph->tensors_size(), in BuildGraph()
/external/tensorflow/tensorflow/lite/experimental/writer/
Dwriter_lib.cc122 tensor_to_written_tensor_.resize(interpreter_->tensors_size(), -1); in ExportTensors()
127 std::vector<bool> tensor_is_temporary(interpreter_->tensors_size(), false); in ExportTensors()
138 for (int tensor_index = 0; tensor_index < interpreter_->tensors_size(); in ExportTensors()
147 for (int tensor_index = 0; tensor_index < interpreter_->tensors_size(); in ExportTensors()
/external/tensorflow/tensorflow/lite/core/
Dsubgraph.cc152 context_->tensors_size = 0; in Subgraph()
177 for (size_t i = 0; i < context_->tensors_size; i++) { in ~Subgraph()
416 if (index < 0 || static_cast<size_t>(index) >= context_->tensors_size) { in CheckTensorIndices()
419 label, context_->tensors_size); in CheckTensorIndices()
590 tensor_index < context_->tensors_size && tensor_index >= 0); in ResizeInputTensor()
787 context_->tensors_size = tensors_.size(); in AddTensors()
830 tensor_index < context_->tensors_size && tensor_index >= 0); in SetTensorParametersReadOnly()
879 tensor_index < context_->tensors_size && tensor_index >= 0); in SetTensorParametersReadWrite()
Dsubgraph.h128 static_cast<size_t>(tensor_index) >= context_->tensors_size) { in tensor()
137 static_cast<size_t>(tensor_index) >= context_->tensors_size) { in tensor()
161 size_t tensors_size() const { return tensors_.size(); } in tensors_size() function
/external/tensorflow/tensorflow/lite/python/interpreter_wrapper/
Dinterpreter_wrapper.cc35 if (i >= interpreter_->tensors_size() || i < 0) { \
38 interpreter_->tensors_size()); \
183 return interpreter_->tensors_size(); in NumTensors()
187 if (!interpreter_ || i >= interpreter_->tensors_size() || i < 0) { in TensorName()
/external/tensorflow/tensorflow/core/kernels/
Dstring_view_variant_wrapper.h56 if (data.tensors_size() != 1 || data.tensors(0).dtype() != DT_STRING) { in Decode()
/external/tensorflow/tensorflow/lite/examples/label_image/
Dlabel_image.cc119 LOG(INFO) << "tensors size: " << interpreter->tensors_size() << "\n"; in RunInference()
124 int t_size = interpreter->tensors_size(); in RunInference()
/external/tensorflow/tensorflow/lite/java/src/test/native/
Dinterpreter_test_jni.cc62 for (size_t i = 0; i < context->tensors_size; ++i) { in Java_org_tensorflow_lite_InterpreterTest_getNativeHandleForDelegate()
/external/tensorflow/tensorflow/core/framework/
Dvariant_tensor_data.h66 int tensors_size() const;
Dvariant_tensor_data.cc27 int VariantTensorData::tensors_size() const { return tensors_.size(); } in tensors_size() function in tensorflow::VariantTensorData
/external/libtextclassifier/annotator/
Dmodel-executor.cc63 if (interpreter->tensors_size() != 2) { in FromBuffer()
/external/tensorflow/tensorflow/lite/c/
Dc_api_internal.h395 size_t tensors_size; member
/external/tensorflow/tensorflow/contrib/lite/
Dnnapi_delegate.cc150 for (size_t i = 0; i < interpreter->tensors_size(); i++) { in addTensorOperands()
722 std::vector<int64_t> tensor_id_to_nnapi_id(interpreter->tensors_size(), in BuildGraph()

12