Searched refs:kWeightsTensor (Results 1 - 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/contrib/lite/kernels/ | ||
H A D | basic_rnn.cc | 35 constexpr int kWeightsTensor = 1; member in namespace:tflite::ops::builtin::rnn 48 &context->tensors[node->inputs->data[kWeightsTensor]]; 91 &context->tensors[node->inputs->data[kWeightsTensor]]; |
H A D | unidirectional_sequence_rnn.cc | 35 constexpr int kWeightsTensor = 1; member in namespace:tflite::ops::builtin::unidirectional_sequence_rnn 48 &context->tensors[node->inputs->data[kWeightsTensor]]; 97 &context->tensors[node->inputs->data[kWeightsTensor]]; |
H A D | fully_connected.cc | 61 constexpr int kWeightsTensor = 1; member in namespace:tflite::ops::builtin::fully_connected 88 TfLiteTensor* filter = GetInput(context, node, kWeightsTensor); 244 TfLiteTensor* filter = GetInput(context, node, kWeightsTensor); |
Completed in 217 milliseconds