HomeSort by relevance Sort by last modified time
    Searched full:kweightstensor (Results 1 - 5 of 5) sorted by null

  /frameworks/ml/nn/common/operations/
RNN.cpp 28 weights_ = GetInput(operation, operands, kWeightsTensor);
52 GetInput(operation, operands, kWeightsTensor);
RNN.h 50 static constexpr int kWeightsTensor = 1; // Optional
  /external/tensorflow/tensorflow/contrib/lite/kernels/
basic_rnn.cc 35 constexpr int kWeightsTensor = 1;
48 &context->tensors[node->inputs->data[kWeightsTensor]];
91 &context->tensors[node->inputs->data[kWeightsTensor]];
unidirectional_sequence_rnn.cc 35 constexpr int kWeightsTensor = 1;
48 &context->tensors[node->inputs->data[kWeightsTensor]];
97 &context->tensors[node->inputs->data[kWeightsTensor]];
fully_connected.cc 61 constexpr int kWeightsTensor = 1;
88 TfLiteTensor* filter = GetInput(context, node, kWeightsTensor);
244 TfLiteTensor* filter = GetInput(context, node, kWeightsTensor);

Completed in 573 milliseconds