Searched refs:kBiasTensor (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/lite/experimental/micro/kernels/ |
| D | depthwise_conv.cc | 35 constexpr int kBiasTensor = 2; variable
66 GetOptionalInputTensor(context, node, kBiasTensor); in CalculateOpData()
164 (NumInputs(node) == 3) ? GetInput(context, node, kBiasTensor) : nullptr; in Eval()
|
| D | fully_connected.cc | 45 constexpr int kBiasTensor = 2; variable
148 const TfLiteTensor* bias = GetOptionalInputTensor(context, node, kBiasTensor); in Eval()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | basic_rnn.cc | 43 constexpr int kBiasTensor = 3; variable
68 const TfLiteTensor* bias = GetInput(context, node, kBiasTensor); in Prepare()
225 const TfLiteTensor* bias = GetInput(context, node, kBiasTensor); in Eval()
|
| D | depthwise_conv.cc | 42 constexpr int kBiasTensor = 2; variable
114 bias = GetInput(context, node, kBiasTensor); in Prepare()
304 (NumInputs(node) == 3) ? GetInput(context, node, kBiasTensor) : nullptr; in Eval()
|
| D | unidirectional_sequence_rnn.cc | 38 constexpr int kBiasTensor = 3; variable
63 const TfLiteTensor* bias = GetInput(context, node, kBiasTensor); in Prepare()
295 const TfLiteTensor* bias = GetInput(context, node, kBiasTensor); in Eval()
|
| D | svdf.cc | 106 constexpr int kBiasTensor = 3; variable
156 const TfLiteTensor* bias = GetOptionalInputTensor(context, node, kBiasTensor); in Prepare()
372 const TfLiteTensor* bias = GetOptionalInputTensor(context, node, kBiasTensor); in Eval()
|
| D | fully_connected.cc | 63 constexpr int kBiasTensor = 2; variable
98 const TfLiteTensor* bias = GetOptionalInputTensor(context, node, kBiasTensor); in Prepare()
470 const TfLiteTensor* bias = GetOptionalInputTensor(context, node, kBiasTensor); in Eval()
|
| /external/tensorflow/tensorflow/lite/experimental/micro/kernels/portable_optimized/ |
| D | depthwise_conv.cc | 35 constexpr int kBiasTensor = 2; variable
69 GetOptionalInputTensor(context, node, kBiasTensor); in CalculateOpData()
395 (NumInputs(node) == 3) ? GetInput(context, node, kBiasTensor) : nullptr; in Eval()
|