Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/lite/kernels/
Dcomplex_support.cc28 static const int kInputTensor = 0; variable
35 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Prepare()
64 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in EvalReal()
96 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in EvalImag()
128 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in EvalAbs()
Dround.cc29 constexpr int kInputTensor = 0; variable
34 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Prepare()
48 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Eval()
Dzeros_like.cc29 constexpr int kInputTensor = 0; variable
36 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Prepare()
48 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Eval()
Dceil.cc28 constexpr int kInputTensor = 0; variable
33 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Prepare()
47 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Eval()
Dneg.cc31 constexpr int kInputTensor = 0; variable
38 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Prepare()
50 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Eval()
Dfloor.cc29 constexpr int kInputTensor = 0; variable
39 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Prepare()
54 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Eval()
Dcumsum.cc29 static const int kInputTensor = 0; variable
37 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Prepare()
56 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Eval()
Dlocal_response_norm.cc35 constexpr int kInputTensor = 0; variable
43 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Prepare()
68 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Eval()
Dmatrix_diag.cc29 constexpr int kInputTensor = 0; variable
36 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Prepare()
126 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Eval()
Dtable.cc25 constexpr int kInputTensor = 0; variable
34 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Prepare()
100 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Eval()
Drank.cc27 constexpr int kInputTensor = 0; variable
35 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Prepare()
Dreverse.cc30 constexpr int kInputTensor = 0; variable
39 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Prepare()
76 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Eval()
Dspace_to_depth.cc38 constexpr int kInputTensor = 0; variable
49 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Prepare()
88 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Eval()
Dcast.cc29 constexpr int kInputTensor = 0; variable
36 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Prepare()
108 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Eval()
Dl2norm.cc38 constexpr int kInputTensor = 0; variable
48 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Prepare()
81 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Eval()
Ddepth_to_space.cc38 constexpr int kInputTensor = 0; variable
49 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Prepare()
92 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Eval()
Dresize_bilinear.cc42 constexpr int kInputTensor = 0; variable
67 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Prepare()
106 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Eval()
Dunpack.cc32 constexpr int kInputTensor = 0; variable
42 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Prepare()
104 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Eval()
Dmatrix_set_diag.cc29 constexpr int kInputTensor = 0; variable
37 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Prepare()
136 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Eval()
Dreverse_sequence.cc31 constexpr int kInputTensor = 0; variable
40 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Prepare()
74 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in ReverseSequenceImpl()
Dreshape.cc30 constexpr int kInputTensor = 0; variable
42 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in ResizeOutput()
162 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Eval()
Daudio_spectrogram.cc36 constexpr int kInputTensor = 0; variable
80 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Prepare()
113 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Eval()
Dshape.cc29 constexpr int kInputTensor = 0; variable
44 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Prepare()
/external/tensorflow/tensorflow/lite/python/testdata/
Dtest_registerer.cc33 constexpr int kInputTensor = 0; variable
41 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Prepare()
54 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kInputTensor, &input)); in Eval()
/external/tensorflow/tensorflow/lite/kernels/perception/
Ddense_image_warp.cc30 constexpr int kInputTensor = 0; variable
94 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Prepare()
127 const TfLiteTensor* input = GetInput(context, node, kInputTensor); in Eval()

123