Home
last modified time | relevance | path

Searched refs:kInputTensor (Results 1 – 25 of 72) sorted by relevance

123

/external/tensorflow/tensorflow/lite/kernels/
Dceil.cc26 constexpr int kInputTensor = 0; variable
30 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Prepare()
41 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Eval()
Dround.cc26 constexpr int kInputTensor = 0; variable
30 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Prepare()
41 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Eval()
Dzeros_like.cc25 constexpr int kInputTensor = 0; variable
31 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Prepare()
40 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Eval()
Dfloor.cc28 constexpr int kInputTensor = 0; variable
37 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Prepare()
49 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Eval()
Dneg.cc28 constexpr int kInputTensor = 0; variable
34 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Prepare()
43 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Eval()
Dshape.cc26 constexpr int kInputTensor = 0; variable
40 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Prepare()
65 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Eval()
Dlocal_response_norm.cc34 constexpr int kInputTensor = 0; variable
41 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Prepare()
63 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Eval()
Dtile.cc30 constexpr int kInputTensor = 0; variable
49 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in ResizeOutput()
144 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Prepare()
167 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Eval()
Dcast.cc32 constexpr int kInputTensor = 0; variable
38 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Prepare()
104 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Eval()
Dmatrix_diag.cc32 constexpr int kInputTensor = 0; variable
38 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Prepare()
123 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Eval()
Dreverse.cc28 constexpr int kInputTensor = 0; variable
36 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Prepare()
68 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Eval()
Drank.cc26 constexpr int kInputTensor = 0; variable
42 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Eval()
Ddepth_to_space.cc35 constexpr int kInputTensor = 0; variable
45 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Prepare()
84 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Eval()
Dspace_to_depth.cc35 constexpr int kInputTensor = 0; variable
45 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Prepare()
80 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Eval()
Dl2norm.cc35 constexpr int kInputTensor = 0; variable
44 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Prepare()
74 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Eval()
Dmatrix_set_diag.cc32 constexpr int kInputTensor = 0; variable
39 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Prepare()
133 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Eval()
Dunpack.cc28 constexpr int kInputTensor = 0; variable
37 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Prepare()
98 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Eval()
Dreverse_sequence.cc28 constexpr int kInputTensor = 0; variable
36 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Prepare()
65 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in ReverseSequenceImpl()
/external/tensorflow/tensorflow/lite/micro/kernels/
Dceil.cc27 constexpr int kInputTensor = 0; variable
31 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Prepare()
46 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Eval()
Dround.cc27 constexpr int kInputTensor = 0; variable
31 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Prepare()
46 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Eval()
Dreshape.cc27 constexpr int kInputTensor = 0; variable
31 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in ReshapeOutput()
76 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Eval()
Dfloor.cc26 constexpr int kInputTensor = 0; variable
30 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Eval()
Dneg.cc27 constexpr int kInputTensor = 0; variable
31 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Eval()
Dlogistic.cc31 constexpr int kInputTensor = 0; variable
39 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Eval()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/
Dtensorrt_test.cc64 const char* kInputTensor = "input"; variable
76 auto input = network->addInput(kInputTensor, nvinfer1::DataType::kFLOAT, in CreateNetwork()
106 const int input_index = engine.getBindingIndex(kInputTensor); in Execute()

123