/external/tensorflow/tensorflow/lite/delegates/coreml/builders/ |
D | convolution_op_builder.cc | 328 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/ |
D | transpose_conv.cc | 37 constexpr int kBiasTensor = 3; variable 96 GetOptionalInputTensor(context, node, kBiasTensor); in CalculateOpData() 206 ? tflite::micro::GetEvalInput(context, node, kBiasTensor) in Eval()
|
D | depthwise_conv.cc | 34 constexpr int kBiasTensor = 2; variable 86 GetOptionalInputTensor(context, node, kBiasTensor); in CalculateOpData() 289 ? tflite::micro::GetEvalInput(context, node, kBiasTensor) in Eval()
|
D | svdf.cc | 38 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/ |
D | basic_rnn.cc | 41 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()
|
D | svdf.cc | 56 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()
|
D | unidirectional_sequence_rnn.cc | 42 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()
|
D | depthwise_conv.cc | 49 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()
|
D | transpose_conv.cc | 51 constexpr int kBiasTensor = 3; variable 275 bias = GetOptionalInputTensor(context, node, kBiasTensor); in Prepare() 578 ? GetOptionalInputTensor(context, node, kBiasTensor) in Eval()
|
D | fully_connected.cc | 124 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/ |
D | fully_connected.cc | 63 constexpr int kBiasTensor = 2; variable 118 const TfLiteTensor* bias = GetOptionalInputTensor(context, node, kBiasTensor); in Prepare() 384 tflite::micro::GetEvalInput(context, node, kBiasTensor); in Eval()
|
D | depthwise_conv.cc | 39 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()
|
D | conv.cc | 38 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/ |
D | svdf.cc | 54 constexpr int kBiasTensor = 3; variable 303 const TfLiteTensor* bias = GetOptionalInputTensor(context, node, kBiasTensor); in Prepare() 432 ? tflite::micro::GetEvalInput(context, node, kBiasTensor) in Eval()
|
D | depthwise_conv.cc | 35 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/ |
D | svdf.cc | 39 constexpr int kBiasTensor = 3; variable 338 const TfLiteTensor* bias = GetOptionalInputTensor(context, node, kBiasTensor); in Prepare() 461 ? tflite::micro::GetEvalInput(context, node, kBiasTensor) in Eval()
|
D | depthwise_conv.cc | 36 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/ |
D | depthwise_conv.cc | 273 constexpr int kBiasTensor = 2; variable 323 GetOptionalInputTensor(context, node, kBiasTensor); in CalculateOpData() 491 ? tflite::micro::GetEvalInput(context, node, kBiasTensor) in Eval()
|