Home
last modified time | relevance | path

Searched refs:TfLiteStatus (Results 1 – 25 of 212) sorted by relevance

123456789

/external/tensorflow/tensorflow/lite/core/
Dsubgraph.h50 TfLiteStatus SetInputs(std::vector<int> inputs);
55 TfLiteStatus SetOutputs(std::vector<int> outputs);
60 TfLiteStatus SetVariables(std::vector<int> variables);
71 TfLiteStatus AddNodeWithParameters(const std::vector<int>& inputs,
81 TfLiteStatus AddTensors(int tensors_to_add,
88 inline TfLiteStatus SetTensorParametersReadOnly(
97 TfLiteStatus SetTensorParametersReadOnly(
106 inline TfLiteStatus SetTensorParametersReadWrite(
113 TfLiteStatus SetTensorParametersReadWrite(int tensor_index, TfLiteType type,
121 TfLiteStatus SetExecutionPlan(const std::vector<int>& new_plan);
[all …]
Dsubgraph.cc27 TfLiteStatus ReportOpError(TfLiteContext* context, const TfLiteNode& node, in ReportOpError()
45 TfLiteStatus ForbiddenContextFunction(TfLiteContext* context, ...) { in ForbiddenContextFunction()
188 TfLiteStatus Subgraph::ReplaceNodeSubsetsWithDelegateKernels( in ReplaceNodeSubsetsWithDelegateKernels()
270 TfLiteStatus Subgraph::ReplaceNodeSubsetsWithDelegateKernels( in ReplaceNodeSubsetsWithDelegateKernels()
354 TfLiteStatus Subgraph::GetExecutionPlan(TfLiteIntArray** execution_plan) { in GetExecutionPlan()
367 TfLiteStatus Subgraph::GetExecutionPlan(struct TfLiteContext* context, in GetExecutionPlan()
373 TfLiteStatus Subgraph::SetInputs(std::vector<int> inputs) { in SetInputs()
380 TfLiteStatus Subgraph::SetOutputs(std::vector<int> outputs) { in SetOutputs()
387 TfLiteStatus Subgraph::SetVariables(std::vector<int> variables) { in SetVariables()
404 TfLiteStatus Subgraph::CheckTensorIndices(const char* label, const int* indices, in CheckTensorIndices()
[all …]
/external/tensorflow/tensorflow/lite/
Darena_planner.h60 TfLiteStatus ResetAllocations() override;
61 TfLiteStatus PlanAllocations() override;
62 TfLiteStatus ExecuteAllocations(int first_node, int last_node) override;
70 TfLiteStatus Commit();
74 TfLiteStatus CalculateAllocations(int first_node, int last_node);
78 TfLiteStatus ResolveTensorAllocation(int tensor_index);
81 TfLiteStatus CalculateTensorAllocation(int tensor_index);
84 TfLiteStatus CalculateTensorDeallocation(int tensor_index);
88 TfLiteStatus CalculateAllocationOfInternalTensors(int node_index);
92 TfLiteStatus CalculateDeallocationOfInternalTensors(int node_index);
Dinterpreter.h125 TfLiteStatus SetInputs(std::vector<int> inputs);
130 TfLiteStatus SetOutputs(std::vector<int> outputs);
135 TfLiteStatus SetVariables(std::vector<int> variables);
146 TfLiteStatus AddNodeWithParameters(const std::vector<int>& inputs,
156 TfLiteStatus AddTensors(int tensors_to_add,
163 TfLiteStatus SetTensorParametersReadOnly(
169 inline TfLiteStatus SetTensorParametersReadOnly(
179 TfLiteStatus SetTensorParametersReadOnly(
188 TfLiteStatus SetTensorParametersReadWrite(int tensor_index, TfLiteType type,
195 inline TfLiteStatus SetTensorParametersReadWrite(
[all …]
Dinterpreter.cc86 TfLiteStatus Interpreter::SetInputs(std::vector<int> inputs) { in SetInputs()
90 TfLiteStatus Interpreter::SetOutputs(std::vector<int> outputs) { in SetOutputs()
94 TfLiteStatus Interpreter::SetVariables(std::vector<int> variables) { in SetVariables()
98 TfLiteStatus Interpreter::AllocateTensors() { in AllocateTensors()
119 TfLiteStatus Interpreter::AddNodeWithParameters( in AddNodeWithParameters()
128 TfLiteStatus Interpreter::ResizeInputTensor(int tensor_index, in ResizeInputTensor()
133 TfLiteStatus Interpreter::Invoke() { in Invoke()
146 TfLiteStatus Interpreter::AddTensors(int tensors_to_add, in AddTensors()
151 TfLiteStatus Interpreter::ResetVariableTensors() { in ResetVariableTensors()
155 TfLiteStatus Interpreter::SetTensorParametersReadOnly( in SetTensorParametersReadOnly()
[all …]
Darena_planner.cc53 TfLiteStatus ArenaPlanner::ResetAllocations() { in ResetAllocations()
63 TfLiteStatus ArenaPlanner::PlanAllocations() { in PlanAllocations()
78 int tensor) -> TfLiteStatus { in PlanAllocations()
89 int node, int tensor) -> TfLiteStatus { in PlanAllocations()
187 TfLiteStatus ArenaPlanner::ExecuteAllocations(int first_node, int last_node) { in ExecuteAllocations()
207 TfLiteStatus ArenaPlanner::Commit() { in Commit()
213 TfLiteStatus ArenaPlanner::CalculateAllocations(int first_node, int last_node) { in CalculateAllocations()
247 TfLiteStatus ArenaPlanner::ResolveTensorAllocation(int tensor_index) { in ResolveTensorAllocation()
264 TfLiteStatus ArenaPlanner::CalculateTensorAllocation(int tensor_index) { in CalculateTensorAllocation()
277 TfLiteStatus ArenaPlanner::CalculateTensorDeallocation(int tensor_index) { in CalculateTensorDeallocation()
[all …]
Dmodel.h193 TfLiteStatus operator()(std::unique_ptr<Interpreter>* interpreter);
194 TfLiteStatus operator()(std::unique_ptr<Interpreter>* interpreter,
198 TfLiteStatus BuildLocalIndexToRegistrationMapping();
199 TfLiteStatus ParseNodes(
202 TfLiteStatus ParseTensors(
206 TfLiteStatus ApplyDelegates(Interpreter* interpreter);
207 TfLiteStatus ParseQuantization(const QuantizationParameters* src_quantization,
Dsimple_memory_arena.h53 TfLiteStatus Allocate(TfLiteContext* context, size_t alignment, size_t size,
56 TfLiteStatus Deallocate(TfLiteContext* context, const ArenaAlloc& alloc);
65 TfLiteStatus Commit(TfLiteContext* context);
67 TfLiteStatus ResolveAlloc(TfLiteContext* context, const ArenaAlloc& alloc,
70 TfLiteStatus Clear();
Dsimple_memory_arena.cc35 TfLiteStatus SimpleMemoryArena::Allocate(TfLiteContext* context, in Allocate()
83 TfLiteStatus SimpleMemoryArena::Deallocate(TfLiteContext* context, in Deallocate()
104 TfLiteStatus SimpleMemoryArena::Commit(TfLiteContext* context) { in Commit()
131 TfLiteStatus SimpleMemoryArena::ResolveAlloc(TfLiteContext* context, in ResolveAlloc()
144 TfLiteStatus SimpleMemoryArena::Clear() { in Clear()
Dmemory_planner.h31 virtual TfLiteStatus PlanAllocations() = 0;
35 virtual TfLiteStatus ExecuteAllocations(int first_node, int last_node) = 0;
40 virtual TfLiteStatus ResetAllocations() = 0;
Dnnapi_delegate.h54 TfLiteStatus BuildGraph(Subgraph* subgraph);
57 TfLiteStatus Invoke(Subgraph* subgraph);
68 TfLiteStatus model_status_ = kTfLiteOk;
/external/tensorflow/tensorflow/lite/kernels/
Delementwise.cc38 TfLiteStatus GenericPrepare(TfLiteContext* context, TfLiteNode* node) { in GenericPrepare()
54 inline TfLiteStatus EvalImpl(TfLiteContext* context, TfLiteNode* node, in EvalImpl()
68 inline TfLiteStatus EvalNumeric(TfLiteContext* context, TfLiteNode* node, in EvalNumeric()
73 inline TfLiteStatus EvalLogical(TfLiteContext* context, TfLiteNode* node, in EvalLogical()
78 TfLiteStatus AbsEval(TfLiteContext* context, TfLiteNode* node) { in AbsEval()
82 TfLiteStatus SinEval(TfLiteContext* context, TfLiteNode* node) { in SinEval()
86 TfLiteStatus CosEval(TfLiteContext* context, TfLiteNode* node) { in CosEval()
90 TfLiteStatus LogEval(TfLiteContext* context, TfLiteNode* node) { in LogEval()
94 TfLiteStatus SqrtEval(TfLiteContext* context, TfLiteNode* node) { in SqrtEval()
98 TfLiteStatus RsqrtEval(TfLiteContext* context, TfLiteNode* node) { in RsqrtEval()
[all …]
Dsparse_to_dense.cc44 TfLiteStatus Resize(TfLiteContext* context, const TfLiteTensor* output_shape, in Resize()
55 TfLiteStatus CheckDimensionsMatch(TfLiteContext* context, in CheckDimensionsMatch()
89 TfLiteStatus GetIndicesVector(TfLiteContext* context, in GetIndicesVector()
132 TfLiteStatus ResizeOutputShape(TfLiteContext* context, in ResizeOutputShape()
146 TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { in Prepare()
191 TfLiteStatus SparseToDenseImpl(TfLiteContext* context, TfLiteNode* node) { in SparseToDenseImpl()
219 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { in Eval()
/external/tensorflow/tensorflow/lite/c/
Dc_api_internal.h40 typedef enum { kTfLiteOk = 0, kTfLiteError = 1 } TfLiteStatus; typedef
60 TfLiteStatus (*Refresh)(struct TfLiteContext* context);
412 TfLiteStatus (*GetExecutionPlan)(struct TfLiteContext* context,
423 TfLiteStatus (*ResizeTensor)(struct TfLiteContext*, TfLiteTensor* tensor,
431 TfLiteStatus (*AddTensors)(struct TfLiteContext*, int tensors_to_add,
436 TfLiteStatus (*GetNodeAndRegistration)(struct TfLiteContext*, int node_index,
442 TfLiteStatus (*ReplaceNodeSubsetsWithDelegateKernels)(
495 TfLiteStatus (*prepare)(TfLiteContext* context, TfLiteNode* node);
499 TfLiteStatus (*invoke)(TfLiteContext* context, TfLiteNode* node);
556 TfLiteStatus (*Prepare)(TfLiteContext* context, TfLiteDelegate* delegate);
[all …]
/external/tensorflow/tensorflow/lite/tools/optimize/
Dsubgraph_quantizer.h38 TfLiteStatus QuantizeOperator(int op_idx);
42 TfLiteStatus QuantizeOpWithBias(BuiltinOperator op_code, OperatorT* op);
46 TfLiteStatus PropagateMinMaxForAvgAndMaxPool(BuiltinOperator op_code,
52 TfLiteStatus AsymmetricQuantizeSoftmax(BuiltinOperator op_code,
56 TfLiteStatus AsymmetricQuantizeInputsAndOutputs(BuiltinOperator op_code,
59 TfLiteStatus AsymmetricQuantizeTensor(BuiltinOperator op_code,
Dquantize_weights.h34 TfLiteStatus QuantizeWeights(flatbuffers::FlatBufferBuilder* builder,
39 TfLiteStatus QuantizeWeights(flatbuffers::FlatBufferBuilder* builder,
49 TfLiteStatus QuantizeWeights(flatbuffers::FlatBufferBuilder* builder,
Dsubgraph_quantizer.cc36 TfLiteStatus AddQuantizationParams(const std::vector<float>& scales, in AddQuantizationParams()
88 TfLiteStatus SymmetricPerChannelQuantizeTensor(ModelT* model, TensorT* tensor, in SymmetricPerChannelQuantizeTensor()
127 TfLiteStatus SymmetricPerChannelBiasQuantize(const TensorT* input_tensor, in SymmetricPerChannelBiasQuantize()
196 TfLiteStatus SubgraphQuantizer::AsymmetricQuantizeTensor( in AsymmetricQuantizeTensor()
222 TfLiteStatus SubgraphQuantizer::QuantizeOpWithBias(BuiltinOperator op_code, in QuantizeOpWithBias()
267 TfLiteStatus SubgraphQuantizer::PropagateMinMaxForAvgAndMaxPool( in PropagateMinMaxForAvgAndMaxPool()
310 TfLiteStatus SubgraphQuantizer::AsymmetricQuantizeSoftmax( in AsymmetricQuantizeSoftmax()
331 TfLiteStatus SubgraphQuantizer::AsymmetricQuantizeInputsAndOutputs( in AsymmetricQuantizeInputsAndOutputs()
359 TfLiteStatus SubgraphQuantizer::QuantizeOperator(int op_idx) { in QuantizeOperator()
/external/tensorflow/tensorflow/lite/testing/kernel_test/
Dinput_generator.h35 TfLiteStatus LoadModel(const string& model_dir);
36 TfLiteStatus ReadInputsFromFile(const string& filename);
37 TfLiteStatus GenerateInput(const string& distribution);
39 TfLiteStatus WriteInputsToFile(const string& filename);
/external/tensorflow/tensorflow/lite/experimental/micro/kernels/
Dfully_connected.cc48 TfLiteStatus CalculateOpData(TfLiteContext* context, in CalculateOpData()
54 TfLiteStatus status = kTfLiteOk; in CalculateOpData()
77 TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { in Prepare()
81 TfLiteStatus EvalQuantized(TfLiteContext* context, TfLiteNode* node, in EvalQuantized()
124 TfLiteStatus EvalFloat(TfLiteContext* context, TfLiteNode* node, in EvalFloat()
142 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { in Eval()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/
Dlogging_op_resolver_test.cc25 TfLiteStatus ConvPrepare(TfLiteContext* context, TfLiteNode* node) { in ConvPrepare()
29 TfLiteStatus ConvEval(TfLiteContext* context, TfLiteNode* node) { in ConvEval()
33 TfLiteStatus AddPrepare(TfLiteContext* context, TfLiteNode* node) { in AddPrepare()
37 TfLiteStatus AddEval(TfLiteContext* context, TfLiteNode* node) { in AddEval()
41 TfLiteStatus WrappingInvoke(TfLiteContext* context, TfLiteNode* node) { in WrappingInvoke()
Dnode_info_delegate.cc25 TfLiteStatus NodeInfoDelegatePrepare(TfLiteContext* context, in NodeInfoDelegatePrepare()
27 if (delegate == nullptr) return TfLiteStatus::kTfLiteError; in NodeInfoDelegatePrepare()
43 TfLiteStatus NodeInfoDelegateObserver::OnDelegatePrepareCalled( in OnDelegatePrepareCalled()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc243 TfLiteStatus AddScalarInt32Operand(int32_t value) { in AddScalarInt32Operand()
247 TfLiteStatus AddScalarFloat32Operand(float value) { in AddScalarFloat32Operand()
251 TfLiteStatus AddVectorInt32Operand(const int32_t* values, in AddVectorInt32Operand()
257 TfLiteStatus AddVectorFloat32Operand(const float* values, in AddVectorFloat32Operand()
263 TfLiteStatus AddPoolingParams(void* data) { in AddPoolingParams()
274 TfLiteStatus AddTensorInput(int tensor_index, bool hybrid_op) { in AddTensorInput()
278 TfLiteStatus AddTensorOutput(int tensor_index) { in AddTensorOutput()
282 TfLiteStatus AddAdditionalFloat32OutputTensor(uint32_t dimension_count) { in AddAdditionalFloat32OutputTensor()
287 TfLiteStatus AddStateFloat32Tensor(int tensor_index, in AddStateFloat32Tensor()
298 TfLiteStatus AddDequantize(int nn_input_index, int lite_index, in AddDequantize()
[all …]
/external/tensorflow/tensorflow/lite/testing/nnapi_tflite_zip_tests/
Dparse_testdata.h41 TfLiteStatus ParseExamples(const char* filename,
46 TfLiteStatus FeedExample(tflite::Interpreter* interpreter, const Example&);
49 TfLiteStatus CheckOutputs(tflite::Interpreter* interpreter, const Example&);
/external/tensorflow/tensorflow/lite/testing/
Dparse_testdata.h38 TfLiteStatus ParseExamples(const char* filename,
43 TfLiteStatus FeedExample(tflite::Interpreter* interpreter, const Example&);
46 TfLiteStatus CheckOutputs(tflite::Interpreter* interpreter, const Example&);
/external/tensorflow/tensorflow/lite/experimental/micro/
Dmicro_interpreter.h40 TfLiteStatus Invoke();
51 TfLiteStatus initialization_status() const { return initialization_status_; } in initialization_status()
61 TfLiteStatus initialization_status_;

123456789