Searched defs:kOutputTensor (Results 1 - 6 of 6) sorted by relevance
| /frameworks/ml/nn/common/operations/ | ||
| H A D | EmbeddingLookup.h | 48 static constexpr int kOutputTensor = 0; member in class:android::nn::EmbeddingLookup |
| H A D | HashtableLookup.h | 49 static constexpr int kOutputTensor = 0; member in class:android::nn::HashtableLookup |
| H A D | LSHProjection.h | 62 static constexpr int kOutputTensor = 0; member in class:android::nn::LSHProjection |
| H A D | RNN.h | 57 static constexpr int kOutputTensor = 1; member in class:android::nn::RNN |
| H A D | SVDF.h | 66 static constexpr int kOutputTensor = 1; member in class:android::nn::SVDF |
| H A D | LSTM.h | 103 static constexpr int kOutputTensor = 3; member in class:android::nn::LSTMCell |
Completed in 455 milliseconds