Home
last modified time | relevance | path

Searched refs:kTfLiteOk (Results 1 – 25 of 64) sorted by relevance

123

/external/tensorflow/tensorflow/contrib/lite/
Dinterpreter_test.cc31 ASSERT_EQ(interpreter.AllocateTensors(), kTfLiteOk); in TEST()
32 ASSERT_EQ(interpreter.Invoke(), kTfLiteOk); in TEST()
38 ASSERT_NE(interpreter.Invoke(), kTfLiteOk); in TEST()
39 ASSERT_EQ(interpreter.AllocateTensors(), kTfLiteOk); in TEST()
40 ASSERT_EQ(interpreter.Invoke(), kTfLiteOk); in TEST()
49 ASSERT_EQ(interpreter.AddTensors(2, &base_index), kTfLiteOk); in TEST()
52 ASSERT_EQ(interpreter.AddTensors(3, &base_index), kTfLiteOk); in TEST()
54 ASSERT_EQ(interpreter.AddTensors(1), kTfLiteOk); in TEST()
65 ASSERT_NE(interpreter.SetInputs({5}), kTfLiteOk); in TEST()
66 ASSERT_NE(interpreter.AllocateTensors(), kTfLiteOk); in TEST()
[all …]
Dmodel_test.cc45 TfLiteStatus dummy_resize(TfLiteContext*, TfLiteNode*) { return kTfLiteOk; } in dummy_resize()
46 TfLiteStatus dummy_invoke(TfLiteContext*, TfLiteNode*) { return kTfLiteOk; } in dummy_invoke()
82 kTfLiteOk); in TEST()
84 ASSERT_NE(InterpreterBuilder(*model, TrivialResolver())(nullptr), kTfLiteOk); in TEST()
95 kTfLiteOk); in TEST()
102 kTfLiteOk); in TEST()
113 kTfLiteOk); in TEST()
126 kTfLiteOk); in TEST()
208 kTfLiteOk); in TEST()
232 ASSERT_NE(interpreter->Invoke(), kTfLiteOk); in TEST()
[all …]
Darena_planner.cc60 return kTfLiteOk; in ResetAllocations()
127 return kTfLiteOk; in PlanAllocations()
142 return kTfLiteOk; in ExecuteAllocations()
148 return kTfLiteOk; in Commit()
182 return kTfLiteOk; in CalculateAllocations()
199 return kTfLiteOk; in ResolveTensorAllocation()
214 return kTfLiteOk; in CalculateTensorAllocation()
222 return kTfLiteOk; in CalculateTensorDeallocation()
235 return kTfLiteOk; in CalculateAllocationOfInternalTensors()
248 return kTfLiteOk; in CalculateDeallocationOfInternalTensors()
Dinterpreter.cc156 return kTfLiteOk; in ReplaceSubgraphsWithDelegateKernels()
170 return kTfLiteOk; in GetExecutionPlan()
185 return kTfLiteOk; in SetInputs()
192 return kTfLiteOk; in SetOutputs()
208 return kTfLiteOk; in CheckTensorIndices()
236 return kTfLiteOk; in BytesRequired()
260 return kTfLiteOk; in AllocateTensors()
304 return kTfLiteOk; in AddNodeWithParameters()
352 return kTfLiteOk; in PrepareOpsStartingAt()
370 return kTfLiteOk; in PrepareOpsAndTensors()
[all …]
Dsimple_memory_arena.cc73 return kTfLiteOk; in Allocate()
90 return kTfLiteOk; in Deallocate()
117 return underlying_buffer_ != nullptr ? kTfLiteOk : kTfLiteError; in Commit()
126 return kTfLiteOk; in ResolveAlloc()
133 return kTfLiteOk; in Clear()
Dmodel.cc123 TfLiteStatus status = kTfLiteOk; in BuildLocalIndexToRegistrationMapping()
578 TfLiteStatus status = kTfLiteOk; in ParseNodes()
624 TfLiteStatus status = kTfLiteOk; in ParseTensors()
682 if (tensor->buffer() == 0) return kTfLiteOk; in ParseTensors()
694 return kTfLiteOk; in ParseTensors()
698 return kTfLiteOk; in ParseTensors()
707 buffer_size, allocation_) != kTfLiteOk) { in ParseTensors()
714 i, type, get_name(tensor), dims, quantization) != kTfLiteOk) { in ParseTensors()
753 if (BuildLocalIndexToRegistrationMapping() != kTfLiteOk) { in operator ()()
778 if ((**interpreter).AddTensors(tensors->Length()) != kTfLiteOk) { in operator ()()
[all …]
Dcontext.h39 typedef enum { kTfLiteOk = 0, kTfLiteError = 1 } TfLiteStatus; enumerator
101 if ((a) != kTfLiteOk) { \
121 if ((status) != kTfLiteOk) { \
Dnnapi_delegate.cc40 if (x != kTfLiteOk) { \
370 if (nn_model_ && nn_compiled_model_) return kTfLiteOk; in BuildGraph()
388 return kTfLiteOk; in BuildGraph()
435 return kTfLiteOk; in Invoke()
Darena_planner_test.cc150 CHECK(planner_->ResetAllocations() == kTfLiteOk); in SetGraph()
151 CHECK(planner_->PlanAllocations() == kTfLiteOk); in SetGraph()
155 CHECK(planner_->ExecuteAllocations(start, end) == kTfLiteOk); in Execute()
214 ASSERT_FALSE(planner_->ExecuteAllocations(0, 10) == kTfLiteOk); in TEST_F()
/external/tensorflow/tensorflow/contrib/lite/kernels/
Dkernel_util_test.cc91 EXPECT_NE(kTfLiteOk, CalculateShapeForBroadcast(&context_, &tensor1_, in TEST_F()
100 EXPECT_EQ(kTfLiteOk, CalculateShapeForBroadcast(&context_, &tensor1_, in TEST_F()
106 EXPECT_EQ(kTfLiteOk, CalculateShapeForBroadcast(&context_, &tensor1_, in TEST_F()
115 EXPECT_EQ(kTfLiteOk, CalculateShapeForBroadcast(&context_, &tensor1_, in TEST_F()
122 EXPECT_EQ(kTfLiteOk, CalculateShapeForBroadcast(&context_, &tensor1_, in TEST_F()
132 EXPECT_EQ(kTfLiteOk, CalculateShapeForBroadcast(&context_, &tensor1_, in TEST_F()
139 EXPECT_EQ(kTfLiteOk, CalculateShapeForBroadcast(&context_, &tensor1_, in TEST_F()
Dactivations.cc136 return kTfLiteOk; in ReluEval()
156 return kTfLiteOk; in Relu1Eval()
174 return kTfLiteOk; in Relu6Eval()
192 return kTfLiteOk; in TanhEval()
227 return kTfLiteOk; in SigmoidEval()
310 return kTfLiteOk; in SoftmaxEval()
314 return kTfLiteOk; in SoftmaxEval()
323 return kTfLiteOk; in SoftmaxEval()
327 return kTfLiteOk; in SoftmaxEval()
Dskip_gram.cc54 return kTfLiteOk; in Prepare()
111 return kTfLiteOk; in Eval()
149 return kTfLiteOk; in Eval()
Dfully_connected.cc130 return kTfLiteOk; in Prepare()
163 return kTfLiteOk; in EvalPie()
206 return kTfLiteOk; in EvalQuantized()
232 return kTfLiteOk; in EvalFloat()
259 return kTfLiteOk; in Eval()
Dsplit.cc45 return kTfLiteOk; in UseDynamicOutputTensors()
68 return kTfLiteOk; in ResizeOutputTensors()
147 return kTfLiteOk; in Eval()
Dtopk_v2.cc63 if (status != kTfLiteOk) { in ResizeOutput()
75 return kTfLiteOk; in ResizeOutput()
181 return kTfLiteOk; in Prepare()
222 return kTfLiteOk; in Eval()
Dmean.cc152 return kTfLiteOk; in InitializeTemporaries()
167 return kTfLiteOk; in Prepare()
217 return kTfLiteOk; in Eval()
Dtest_util.cc191 CHECK(interpreter_->ResizeInputTensor(input_idx, shape) == kTfLiteOk); in BuildInterpreter()
193 CHECK(interpreter_->AllocateTensors() == kTfLiteOk) in BuildInterpreter()
197 void SingleOpModel::Invoke() { CHECK(interpreter_->Invoke() == kTfLiteOk); } in Invoke()
Dconv.cc200 if (output_status != kTfLiteOk) return output_status; in Prepare()
248 if (im2col_status != kTfLiteOk) return im2col_status; in Prepare()
280 if (hwcn_weights_status != kTfLiteOk) return hwcn_weights_status; in Prepare()
287 return kTfLiteOk; in Prepare()
436 return kTfLiteOk; in Eval()
Dresize_bilinear.cc71 return kTfLiteOk; in Prepare()
109 return kTfLiteOk; in Eval()
Dkernel_util.cc41 return kTfLiteOk; in GetQuantizedConvolutionMultipler()
110 return kTfLiteOk; in CalculateShapeForBroadcast()
/external/tensorflow/tensorflow/contrib/lite/java/src/main/native/
Dnativeinterpreterwrapper_jni.cc149 return kTfLiteOk; in checkInputs()
160 if (status != kTfLiteOk) { in resizeInputs()
166 return kTfLiteOk; in resizeInputs()
200 return kTfLiteOk; in setInputs()
343 if (status != kTfLiteOk) { in Java_org_tensorflow_lite_NativeInterpreterWrapper_createInterpreter()
367 if (status != kTfLiteOk) return nullptr; in Java_org_tensorflow_lite_NativeInterpreterWrapper_run()
370 if (status != kTfLiteOk) { in Java_org_tensorflow_lite_NativeInterpreterWrapper_run()
377 if (status != kTfLiteOk) { in Java_org_tensorflow_lite_NativeInterpreterWrapper_run()
386 if (status != kTfLiteOk) return nullptr; in Java_org_tensorflow_lite_NativeInterpreterWrapper_run()
388 if (interpreter->Invoke() != kTfLiteOk) { in Java_org_tensorflow_lite_NativeInterpreterWrapper_run()
[all …]
/external/tensorflow/tensorflow/contrib/lite/examples/label_image/
Dlabel_image.cc70 return kTfLiteOk; in ReadLabelsFile()
138 if (interpreter->AllocateTensors() != kTfLiteOk) { in RunInference()
172 if (interpreter->Invoke() != kTfLiteOk) { in RunInference()
208 if (ReadLabelsFile(s->labels_file_name, &labels, &label_count) != kTfLiteOk) in RunInference()
/external/tensorflow/tensorflow/contrib/lite/models/smartreply/ops/
Dextract_feature.cc75 return kTfLiteOk; in Prepare()
105 return kTfLiteOk; in Eval()
/external/tensorflow/tensorflow/contrib/lite/testing/
Dtflite_driver.cc125 if (interpreter_->AllocateTensors() != kTfLiteOk) { in AllocateTensors()
161 id, testing::Split<int>(csv_values, ",")) != kTfLiteOk) { in ReshapeTensor()
233 if (interpreter_->Invoke() != kTfLiteOk) { in Invoke()
Dparse_testdata.cc125 return kTfLiteOk; in ParseExamples()
148 return kTfLiteOk; in ParseExamples()
183 return kTfLiteOk; in FeedExample()
247 return kTfLiteOk; in CheckOutputs()

123