/external/tensorflow/tensorflow/contrib/lite/kernels/ |
D | kernel_util.h | 27 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()
|
D | activations.cc | 55 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()
|
D | pooling.cc | 65 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()
|
D | split.cc | 31 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()
|
D | exp.cc | 35 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()
|
D | mul.cc | 54 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()
|
D | squeeze.cc | 29 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()
|
D | fully_connected.cc | 78 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()
|
D | add.cc | 54 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()
|
D | div.cc | 40 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()
|
D | sub.cc | 40 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()
|
D | mean.cc | 35 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()
|
D | transpose.cc | 35 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()
|
D | batch_to_space_nd.cc | 37 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()
|
D | pad.cc | 37 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()
|
D | space_to_batch_nd.cc | 37 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()
|
D | depthwise_conv.cc | 75 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/ |
D | interpreter.h | 194 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>>
|
D | model_test.cc | 45 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()
|
D | context.h | 237 } TfLiteNode; typedef 283 TfLiteNode** node, 322 TfLiteStatus (*prepare)(TfLiteContext* context, TfLiteNode* node); 326 TfLiteStatus (*invoke)(TfLiteContext* context, TfLiteNode* node);
|
D | interpreter_test.cc | 358 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 …]
|
D | arena_planner.cc | 83 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()
|
D | interpreter.cc | 55 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()
|
D | graph_info_test.cc | 44 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_;
|
D | optional_debug_tools.cc | 92 const std::pair<TfLiteNode, TfLiteRegistration>* node_and_reg = in PrintInterpreterState() 94 const TfLiteNode& node = node_and_reg->first; in PrintInterpreterState()
|