Home
last modified time | relevance | path

Searched refs:TfLiteNode (Results 1 – 25 of 55) sorted by relevance

123

/external/tensorflow/tensorflow/contrib/lite/kernels/
Dkernel_util.h27 inline TfLiteTensor* GetInput(TfLiteContext* context, TfLiteNode* node, in GetInput()
31 inline TfLiteTensor* GetOutput(TfLiteContext* context, TfLiteNode* node, in GetOutput()
35 inline int NumInputs(const TfLiteNode* node) { return node->inputs->size; } in NumInputs()
36 inline int NumOutputs(const TfLiteNode* node) { return node->outputs->size; } in NumOutputs()
47 const TfLiteNode* node, int index) { in GetOptionalInputTensor()
Dactivations.cc55 TfLiteStatus GenericPrepare(TfLiteContext* context, TfLiteNode* node) { in GenericPrepare()
66 TfLiteStatus SigmoidPrepare(TfLiteContext* context, TfLiteNode* node) { in SigmoidPrepare()
96 TfLiteStatus SoftmaxPrepare(TfLiteContext* context, TfLiteNode* node) { in SoftmaxPrepare()
126 TfLiteStatus ReluEval(TfLiteContext* context, TfLiteNode* node) { in ReluEval()
144 TfLiteStatus Relu1Eval(TfLiteContext* context, TfLiteNode* node) { in Relu1Eval()
164 TfLiteStatus Relu6Eval(TfLiteContext* context, TfLiteNode* node) { in Relu6Eval()
182 TfLiteStatus TanhEval(TfLiteContext* context, TfLiteNode* node) { in TanhEval()
201 TfLiteStatus SigmoidEval(TfLiteContext* context, TfLiteNode* node) { in SigmoidEval()
297 TfLiteStatus SoftmaxEval(TfLiteContext* context, TfLiteNode* node) { in SoftmaxEval()
Dpooling.cc65 TfLiteStatus GenericPrepare(TfLiteContext* context, TfLiteNode* node) { in GenericPrepare()
123 void AverageEvalFloat(TfLiteContext* context, TfLiteNode* node, in AverageEvalFloat()
144 void AverageEvalQuantized(TfLiteContext* context, TfLiteNode* node, in AverageEvalQuantized()
167 void MaxEvalFloat(TfLiteContext* context, TfLiteNode* node, in MaxEvalFloat()
188 void MaxEvalQuantized(TfLiteContext* context, TfLiteNode* node, in MaxEvalQuantized()
211 void L2EvalFloat(TfLiteContext* context, TfLiteNode* node, in L2EvalFloat()
234 TfLiteStatus AverageEval(TfLiteContext* context, TfLiteNode* node) { in AverageEval()
256 TfLiteStatus MaxEval(TfLiteContext* context, TfLiteNode* node) { in MaxEval()
277 TfLiteStatus L2Eval(TfLiteContext* context, TfLiteNode* node) { in L2Eval()
Dsplit.cc31 OpContext(TfLiteContext* context, TfLiteNode* node) { in OpContext()
41 TfLiteStatus UseDynamicOutputTensors(TfLiteContext* context, TfLiteNode* node) { in UseDynamicOutputTensors()
48 TfLiteStatus ResizeOutputTensors(TfLiteContext* context, TfLiteNode* node, in ResizeOutputTensors()
71 TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { in Prepare()
95 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { in Eval()
Dexp.cc35 ExpContext(TfLiteContext* context, TfLiteNode* node) { in ExpContext()
43 TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { in Prepare()
54 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { in Eval()
Dmul.cc54 TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { in Prepare()
81 void EvalFloat(TfLiteContext* context, TfLiteNode* node, in EvalFloat()
110 void EvalQuantized(TfLiteContext* context, TfLiteNode* node, in EvalQuantized()
148 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { in Eval()
Dsqueeze.cc29 SqueezeContext(TfLiteContext* context, TfLiteNode* node) { in SqueezeContext()
39 TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { in Prepare()
81 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { in Eval()
Dfully_connected.cc78 TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { in Prepare()
133 TfLiteStatus EvalPie(TfLiteContext* context, TfLiteNode* node, in EvalPie()
178 TfLiteStatus EvalQuantized(TfLiteContext* context, TfLiteNode* node, in EvalQuantized()
210 TfLiteStatus EvalFloat(TfLiteContext* context, TfLiteNode* node, in EvalFloat()
238 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { in Eval()
Dadd.cc54 TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { in Prepare()
81 void EvalAddFloat(TfLiteContext* context, TfLiteNode* node, in EvalAddFloat()
110 void EvalAddQuantized(TfLiteContext* context, TfLiteNode* node, in EvalAddQuantized()
163 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { in Eval()
Ddiv.cc40 TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { in Prepare()
62 void EvalDivFloat(TfLiteContext* context, TfLiteNode* node, in EvalDivFloat()
82 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { in Eval()
Dsub.cc40 TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { in Prepare()
62 void EvalSubFloat(TfLiteContext* context, TfLiteNode* node, in EvalSubFloat()
82 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { in Eval()
Dmean.cc35 MeanContext(TfLiteContext* context, TfLiteNode* node) { in MeanContext()
133 TfLiteStatus InitializeTemporaries(TfLiteContext* context, TfLiteNode* node, in InitializeTemporaries()
155 TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { in Prepare()
176 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { in Eval()
Dtranspose.cc35 TransposeContext(TfLiteContext* context, TfLiteNode* node) { in TransposeContext()
68 TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { in Prepare()
87 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { in Eval()
Dbatch_to_space_nd.cc37 BatchToSpaceNDContext(TfLiteContext* context, TfLiteNode* node) { in BatchToSpaceNDContext()
95 TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { in Prepare()
113 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { in Eval()
Dpad.cc37 PadContext(TfLiteContext* context, TfLiteNode* node) { in PadContext()
78 TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { in Prepare()
98 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { in Eval()
Dspace_to_batch_nd.cc37 SpaceToBatchNDContext(TfLiteContext* context, TfLiteNode* node) { in SpaceToBatchNDContext()
90 TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { in Prepare()
108 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { in Eval()
Ddepthwise_conv.cc75 TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { in Prepare()
170 void EvalFloat(TfLiteContext* context, TfLiteNode* node, in EvalFloat()
197 void EvalQuantized(TfLiteContext* context, TfLiteNode* node, in EvalQuantized()
227 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { in Eval()
/external/tensorflow/tensorflow/contrib/lite/
Dinterpreter.h194 const std::pair<TfLiteNode, TfLiteRegistration>* node_and_registration( in node_and_registration()
279 TfLiteStatus OpPrepare(const TfLiteRegistration& op_reg, TfLiteNode* node) { in OpPrepare()
285 TfLiteStatus OpInvoke(const TfLiteRegistration& op_reg, TfLiteNode* node) { in OpInvoke()
354 TfLiteStatus GetNodeAndRegistration(int node_index, TfLiteNode** node,
360 int node_index, TfLiteNode** node,
380 std::vector<std::pair<TfLiteNode, TfLiteRegistration>>
Dmodel_test.cc45 TfLiteStatus dummy_resize(TfLiteContext*, TfLiteNode*) { return kTfLiteOk; } in dummy_resize() argument
46 TfLiteStatus dummy_invoke(TfLiteContext*, TfLiteNode*) { return kTfLiteOk; } in dummy_invoke() argument
160 const std::pair<TfLiteNode, TfLiteRegistration>* node_and_reg0 = in TEST()
163 const TfLiteNode& node0 = node_and_reg0->first; in TEST()
179 const std::pair<TfLiteNode, TfLiteRegistration>* node_and_reg1 = in TEST()
182 const TfLiteNode& node1 = node_and_reg1->first; in TEST()
Dcontext.h237 } TfLiteNode; typedef
283 TfLiteNode** node,
322 TfLiteStatus (*prepare)(TfLiteContext* context, TfLiteNode* node);
326 TfLiteStatus (*invoke)(TfLiteContext* context, TfLiteNode* node);
Dinterpreter_test.cc358 reg.prepare = [](TfLiteContext* context, TfLiteNode* node) { in TEST()
385 reg.invoke = [](TfLiteContext* context, TfLiteNode* node) { in TEST()
441 reg_copy.invoke = [](TfLiteContext* context, TfLiteNode* node) { in TEST()
453 reg_len.prepare = [](TfLiteContext* context, TfLiteNode* node) { in TEST()
459 reg_len.invoke = [](TfLiteContext* context, TfLiteNode* node) { in TEST()
509 reg.prepare = [](TfLiteContext* context, TfLiteNode* node) { in TEST()
515 reg.invoke = [](TfLiteContext* context, TfLiteNode* node) { in TEST()
592 reg.prepare = [](TfLiteContext* context, TfLiteNode* node) { in CopyOpRegistration()
600 reg.invoke = [](TfLiteContext* context, TfLiteNode* node) { in CopyOpRegistration()
699 reg.prepare = [](TfLiteContext* context, TfLiteNode* node) { in AddOpRegistration()
[all …]
Darena_planner.cc83 const TfLiteNode& node = graph_info_->node(i); in PlanAllocations()
102 const TfLiteNode& node = graph_info_->node(i); in PlanAllocations()
228 const TfLiteNode& node = graph_info_->node(node_index); in CalculateAllocationOfInternalTensors()
241 const TfLiteNode& node = graph_info_->node(node_index); in CalculateDeallocationOfInternalTensors()
Dinterpreter.cc55 const TfLiteNode& node(size_t index) const override { in node()
95 TfLiteNode& node = nodeAndReg.first; in ~Interpreter()
283 TfLiteNode& node = node_and_reg.first; in AddNodeWithParameters()
336 TfLiteNode& node = nodes_and_registration_[node_index].first; in PrepareOpsStartingAt()
413 TfLiteNode& node = nodes_and_registration_[node_index].first; in Invoke()
471 int node_index, TfLiteNode** node, TfLiteRegistration** registration) { in GetNodeAndRegistration()
480 struct TfLiteContext* context, int node_index, TfLiteNode** node, in GetNodeAndRegistration()
Dgraph_info_test.cc44 const TfLiteNode& node(size_t index) const override { return nodes_[index]; } in node()
51 nodes_.push_back(TfLiteNode()); in AddNode()
52 TfLiteNode& node = nodes_.back(); in AddNode()
66 std::vector<TfLiteNode> nodes_;
Doptional_debug_tools.cc92 const std::pair<TfLiteNode, TfLiteRegistration>* node_and_reg = in PrintInterpreterState()
94 const TfLiteNode& node = node_and_reg->first; in PrintInterpreterState()

123