Home	Sort by relevance Sort by last modified time

Full Search Definition Symbol File Path History   | | Help

    Searched refs:kOutputTensor (Results 1 - 13 of 13) sorted by null

/frameworks/ml/nn/common/operations/
EmbeddingLookup.h	48 static constexpr int kOutputTensor = 0;
HashtableLookup.h	49 static constexpr int kOutputTensor = 0;
EmbeddingLookup.cpp	31 output_ = GetOutput(operation, operands, kOutputTensor);
LSHProjection.h	62 static constexpr int kOutputTensor = 0;
RNN.h	57 static constexpr int kOutputTensor = 1;
SVDF.h	66 static constexpr int kOutputTensor = 1;
HashtableLookup.cpp	40 output_ = GetOutput(operation, operands, kOutputTensor);
LSTM.h	102 static constexpr int kOutputTensor = 3;
LSHProjection.cpp	35 output_ = GetOutput(operation, operands, kOutputTensor);
RNN.cpp	37 output_ = GetOutput(operation, operands, kOutputTensor);
SVDF.cpp	57 output_ = GetOutput(operation, operands, kOutputTensor);
LSTM.cpp	77 output_ = GetOutput(operation, operands, kOutputTensor);
/frameworks/ml/nn/common/
CpuExecutor.cpp	[all...]

Completed in 150 milliseconds