Home
last modified time | relevance | path

Searched refs:kBiasTensor (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/coreml/builders/
Dconvolution_op_builder.cc328 const int kBiasTensor = 2; // Only used for non-TransposeConv in IsConvolutionOpSupported() local
345 if (node->inputs->size >= kBiasTensor && in IsConvolutionOpSupported()
346 !IsConstantTensor(GetInput(context, node, kBiasTensor))) { in IsConvolutionOpSupported()
/external/tensorflow/tensorflow/lite/micro/kernels/
Dtranspose_conv.cc37 constexpr int kBiasTensor = 3; variable
96 GetOptionalInputTensor(context, node, kBiasTensor); in CalculateOpData()
206 ? tflite::micro::GetEvalInput(context, node, kBiasTensor) in Eval()
Ddepthwise_conv.cc34 constexpr int kBiasTensor = 2; variable
86 GetOptionalInputTensor(context, node, kBiasTensor); in CalculateOpData()
289 ? tflite::micro::GetEvalInput(context, node, kBiasTensor) in Eval()
Dsvdf.cc38 constexpr int kBiasTensor = 3; variable
215 const TfLiteTensor* bias = GetOptionalInputTensor(context, node, kBiasTensor); in Prepare()
346 ? tflite::micro::GetEvalInput(context, node, kBiasTensor) in Eval()
/external/tensorflow/tensorflow/lite/kernels/
Dbasic_rnn.cc41 constexpr int kBiasTensor = 3; variable
73 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kBiasTensor, &bias)); in Prepare()
293 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kBiasTensor, &bias)); in Eval()
Dsvdf.cc56 constexpr int kBiasTensor = 3; variable
109 const TfLiteTensor* bias = GetOptionalInputTensor(context, node, kBiasTensor); in Prepare()
294 const TfLiteTensor* bias = GetOptionalInputTensor(context, node, kBiasTensor); in Eval()
Dunidirectional_sequence_rnn.cc42 constexpr int kBiasTensor = 3; variable
74 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kBiasTensor, &bias)); in Prepare()
368 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kBiasTensor, &bias)); in Eval()
Ddepthwise_conv.cc49 constexpr int kBiasTensor = 2; variable
142 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kBiasTensor, &bias)); in Prepare()
535 (NumInputs(node) == 3) ? GetInput(context, node, kBiasTensor) : nullptr; in EvalImpl()
Dtranspose_conv.cc51 constexpr int kBiasTensor = 3; variable
275 bias = GetOptionalInputTensor(context, node, kBiasTensor); in Prepare()
578 ? GetOptionalInputTensor(context, node, kBiasTensor) in Eval()
Dfully_connected.cc124 constexpr int kBiasTensor = 2; variable
208 ? GetOptionalInputTensor(context, node, kBiasTensor) in PrepareImpl()
921 ? GetOptionalInputTensor(context, node, kBiasTensor) in Eval()
/external/tensorflow/tensorflow/lite/micro/kernels/arc_mli/
Dfully_connected.cc63 constexpr int kBiasTensor = 2; variable
118 const TfLiteTensor* bias = GetOptionalInputTensor(context, node, kBiasTensor); in Prepare()
384 tflite::micro::GetEvalInput(context, node, kBiasTensor); in Eval()
Ddepthwise_conv.cc39 constexpr int kBiasTensor = 2; variable
121 const TfLiteTensor* bias = GetOptionalInputTensor(context, node, kBiasTensor); in CalculateOpData()
154 const TfLiteTensor* bias = GetOptionalInputTensor(context, node, kBiasTensor); in Prepare()
547 ? tflite::micro::GetEvalInput(context, node, kBiasTensor) in Eval()
Dconv.cc38 constexpr int kBiasTensor = 2; variable
131 const TfLiteTensor* bias = GetOptionalInputTensor(context, node, kBiasTensor); in CalculateOpData()
163 const TfLiteTensor* bias = GetOptionalInputTensor(context, node, kBiasTensor); in Prepare()
521 tflite::micro::GetEvalInput(context, node, kBiasTensor); in Eval()
/external/tensorflow/tensorflow/lite/micro/kernels/cmsis_nn/
Dsvdf.cc54 constexpr int kBiasTensor = 3; variable
303 const TfLiteTensor* bias = GetOptionalInputTensor(context, node, kBiasTensor); in Prepare()
432 ? tflite::micro::GetEvalInput(context, node, kBiasTensor) in Eval()
Ddepthwise_conv.cc35 constexpr int kBiasTensor = 2; variable
90 GetOptionalInputTensor(context, node, kBiasTensor); in CalculateOpData()
438 ? tflite::micro::GetEvalInput(context, node, kBiasTensor) in Eval()
/external/tensorflow/tensorflow/lite/micro/kernels/xtensa/
Dsvdf.cc39 constexpr int kBiasTensor = 3; variable
338 const TfLiteTensor* bias = GetOptionalInputTensor(context, node, kBiasTensor); in Prepare()
461 ? tflite::micro::GetEvalInput(context, node, kBiasTensor) in Eval()
Ddepthwise_conv.cc36 constexpr int kBiasTensor = 2; variable
332 GetOptionalInputTensor(context, node, kBiasTensor); in CalculateOpData()
470 ? tflite::micro::GetEvalInput(context, node, kBiasTensor) in Eval()
/external/tensorflow/tensorflow/lite/micro/kernels/vexriscv/
Ddepthwise_conv.cc273 constexpr int kBiasTensor = 2; variable
323 GetOptionalInputTensor(context, node, kBiasTensor); in CalculateOpData()
491 ? tflite::micro::GetEvalInput(context, node, kBiasTensor) in Eval()