Home
last modified time | relevance | path

Searched refs:RunTimeOperandInfo (Results 1 – 20 of 20) sorted by relevance

/frameworks/ml/nn/common/operations/
DBidirectionalSequenceLSTM.h33 struct RunTimeOperandInfo;
37 BidirectionalSequenceLSTM(const hal::Operation& operation, RunTimeOperandInfo* operands);
39 bool Prepare(const hal::Operation& operation, RunTimeOperandInfo* operands,
163 const RunTimeOperandInfo* input_;
165 const RunTimeOperandInfo* aux_input_;
166 const RunTimeOperandInfo* fw_aux_input_to_input_weights_;
167 const RunTimeOperandInfo* fw_aux_input_to_forget_weights_;
168 const RunTimeOperandInfo* fw_aux_input_to_cell_weights_;
169 const RunTimeOperandInfo* fw_aux_input_to_output_weights_;
170 const RunTimeOperandInfo* bw_aux_input_to_input_weights_;
[all …]
DLSTM.h46 struct RunTimeOperandInfo;
51 LSTMCell(const hal::Operation& operation, RunTimeOperandInfo* operands);
53 bool Prepare(const hal::Operation& operation, RunTimeOperandInfo* operands, Shape* scratchShape,
185 const RunTimeOperandInfo* input_, const RunTimeOperandInfo* input_to_input_weights,
186 const RunTimeOperandInfo* input_to_forget_weights,
187 const RunTimeOperandInfo* input_to_cell_weights,
188 const RunTimeOperandInfo* input_to_output_weights,
189 const RunTimeOperandInfo* recurrent_to_input_weights,
190 const RunTimeOperandInfo* recurrent_to_forget_weights,
191 const RunTimeOperandInfo* recurrent_to_cell_weights,
[all …]
DQuantizedLSTM.h27 struct RunTimeOperandInfo;
31 QuantizedLSTMCell(const hal::Operation& operation, RunTimeOperandInfo* operands);
33 static bool prepare(const hal::Operation& operation, RunTimeOperandInfo* operands,
65 const RunTimeOperandInfo* input_;
67 const RunTimeOperandInfo* inputToInputWeights_;
68 const RunTimeOperandInfo* inputToForgetWeights_;
69 const RunTimeOperandInfo* inputToCellWeights_;
70 const RunTimeOperandInfo* inputToOutputWeights_;
72 const RunTimeOperandInfo* recurrentToInputWeights_;
73 const RunTimeOperandInfo* recurrentToForgetWeights_;
[all …]
DLSHProjection.h34 struct RunTimeOperandInfo;
39 LSHProjection(const hal::Operation& operation, RunTimeOperandInfo* operands);
41 static bool Prepare(const hal::Operation& operation, RunTimeOperandInfo* operands,
57 const RunTimeOperandInfo* hash_;
58 const RunTimeOperandInfo* input_;
59 const RunTimeOperandInfo* weight_;
61 RunTimeOperandInfo* output_;
65 int runningSignBit(const RunTimeOperandInfo* input, const RunTimeOperandInfo* weight, T seed);
68 void SparseLshProjection(LSHProjectionType type, const RunTimeOperandInfo* hash,
69 const RunTimeOperandInfo* input, const RunTimeOperandInfo* weight,
[all …]
DLSHProjection.cpp34 LSHProjection::LSHProjection(const Operation& operation, RunTimeOperandInfo* operands) { in LSHProjection()
45 bool LSHProjection::Prepare(const Operation& operation, RunTimeOperandInfo* operands, in Prepare()
55 const RunTimeOperandInfo* hash = GetInput(operation, operands, kHashTensor); in Prepare()
60 const RunTimeOperandInfo* input = GetInput(operation, operands, kInputTensor); in Prepare()
73 RunTimeOperandInfo* weight = GetInput(operation, operands, kWeightTensor); in Prepare()
97 int runningSignBit(const RunTimeOperandInfo* input, const RunTimeOperandInfo* weight, float seed) { in runningSignBit()
126 void SparseLshProjection(LSHProjectionType type, const RunTimeOperandInfo* hash, in SparseLshProjection()
127 const RunTimeOperandInfo* input, const RunTimeOperandInfo* weight, in SparseLshProjection()
147 void DenseLshProjection(const RunTimeOperandInfo* hash, const RunTimeOperandInfo* input, in DenseLshProjection()
148 const RunTimeOperandInfo* weight, int32_t* out_buf) { in DenseLshProjection()
[all …]
DSVDF.h36 struct RunTimeOperandInfo;
41 SVDF(const hal::Operation& operation, RunTimeOperandInfo* operands);
43 static bool Prepare(const hal::Operation& operation, RunTimeOperandInfo* operands,
65 const RunTimeOperandInfo* input_;
66 const RunTimeOperandInfo* weights_feature_;
67 const RunTimeOperandInfo* weights_time_;
68 const RunTimeOperandInfo* bias_;
69 const RunTimeOperandInfo* state_in_;
71 RunTimeOperandInfo* state_out_;
72 RunTimeOperandInfo* output_;
DRNN.h28 struct RunTimeOperandInfo;
33 RNN(const hal::Operation& operation, RunTimeOperandInfo* operands);
35 static bool Prepare(const hal::Operation& operation, RunTimeOperandInfo* operands,
67 const RunTimeOperandInfo* input_;
68 const RunTimeOperandInfo* weights_;
69 const RunTimeOperandInfo* recurrent_weights_;
70 const RunTimeOperandInfo* bias_;
71 const RunTimeOperandInfo* hidden_state_in_;
73 RunTimeOperandInfo* hidden_state_out_;
74 RunTimeOperandInfo* output_;
DHashtableLookup.h27 struct RunTimeOperandInfo;
31 HashtableLookup(const hal::Operation& operation, RunTimeOperandInfo* operands);
43 const RunTimeOperandInfo* lookup_;
44 const RunTimeOperandInfo* key_;
45 const RunTimeOperandInfo* value_;
47 RunTimeOperandInfo* output_;
48 RunTimeOperandInfo* hits_;
DMultinomial.h31 struct RunTimeOperandInfo;
36 Multinomial(const hal::Operation& operation, RunTimeOperandInfo* operands);
38 static bool Prepare(const hal::Operation& operation, RunTimeOperandInfo* operands,
51 RunTimeOperandInfo* input_;
53 RunTimeOperandInfo* random_seeds_;
55 RunTimeOperandInfo* output_;
DEmbeddingLookup.h27 struct RunTimeOperandInfo;
31 EmbeddingLookup(const hal::Operation& operation, RunTimeOperandInfo* operands);
41 const RunTimeOperandInfo* value_;
42 const RunTimeOperandInfo* lookup_;
44 RunTimeOperandInfo* output_;
DMultinomial.cpp43 inline T* GetBuffer(RunTimeOperandInfo* operand) { in GetBuffer()
48 inline const T* GetBuffer(const RunTimeOperandInfo* operand) { in GetBuffer()
54 Multinomial::Multinomial(const Operation& operation, RunTimeOperandInfo* operands) { in Multinomial()
63 bool Multinomial::Prepare(const Operation& operation, RunTimeOperandInfo* operands, in Prepare()
69 const RunTimeOperandInfo* input = GetInput(operation, operands, Multinomial::kInputTensor); in Prepare()
DLSTM.cpp38 inline T* GetBuffer(RunTimeOperandInfo* operand) { in GetBuffer()
43 inline const T* GetBuffer(const RunTimeOperandInfo* operand) { in GetBuffer()
48 inline const T* GetOptionalBuffer(const RunTimeOperandInfo* operand) { in GetOptionalBuffer()
54 LSTMCell::LSTMCell(const Operation& operation, RunTimeOperandInfo* operands) { in LSTMCell()
115 static RunTimeOperandInfo no_value; in LSTMCell()
133 const RunTimeOperandInfo* input_, const RunTimeOperandInfo* input_to_input_weights, in CheckInputTensorDimensions()
134 const RunTimeOperandInfo* input_to_forget_weights, in CheckInputTensorDimensions()
135 const RunTimeOperandInfo* input_to_cell_weights, in CheckInputTensorDimensions()
136 const RunTimeOperandInfo* input_to_output_weights, in CheckInputTensorDimensions()
137 const RunTimeOperandInfo* recurrent_to_input_weights, in CheckInputTensorDimensions()
[all …]
DSVDF.cpp33 SVDF::SVDF(const Operation& operation, RunTimeOperandInfo* operands) { in SVDF()
51 bool SVDF::Prepare(const Operation& operation, RunTimeOperandInfo* operands, Shape* stateShape, in Prepare()
74 const RunTimeOperandInfo* input = GetInput(operation, operands, SVDF::kInputTensor); in Prepare()
75 const RunTimeOperandInfo* weights_feature = in Prepare()
77 const RunTimeOperandInfo* weights_time = in Prepare()
91 const RunTimeOperandInfo* bias = GetInput(operation, operands, kBiasTensor); in Prepare()
DRNN.cpp34 RNN::RNN(const Operation& operation, RunTimeOperandInfo* operands) { in RNN()
49 bool RNN::Prepare(const Operation& operation, RunTimeOperandInfo* operands, Shape* hiddenStateShape, in Prepare()
57 const RunTimeOperandInfo* input = GetInput(operation, operands, kInputTensor); in Prepare()
58 const RunTimeOperandInfo* input_weights = GetInput(operation, operands, kWeightsTensor); in Prepare()
59 const RunTimeOperandInfo* recurrent_weights = in Prepare()
61 const RunTimeOperandInfo* bias = GetInput(operation, operands, kBiasTensor); in Prepare()
DQuantizedLSTM.cpp40 inline T* GetBuffer(RunTimeOperandInfo* operand) { in GetBuffer()
45 inline const T* GetBuffer(const RunTimeOperandInfo* operand) { in GetBuffer()
209 void assignWeightsSubmatrix(const RunTimeOperandInfo* submatrix, const int32_t offset_row, in assignWeightsSubmatrix()
223 QuantizedLSTMCell::QuantizedLSTMCell(const Operation& operation, RunTimeOperandInfo* operands) { in QuantizedLSTMCell()
248 bool QuantizedLSTMCell::prepare(const Operation& operation, RunTimeOperandInfo* operands, in prepare()
269 auto checkWeightsShape = [&](const RunTimeOperandInfo* weights, uint32_t columns) -> bool { in prepare()
301 auto checkBiasShape = [&](const RunTimeOperandInfo* bias) -> bool { in prepare()
DEmbeddingLookup.cpp32 EmbeddingLookup::EmbeddingLookup(const Operation& operation, RunTimeOperandInfo* operands) { in EmbeddingLookup()
DBidirectionalSequenceLSTM.cpp38 inline T* GetBuffer(RunTimeOperandInfo* operand) { in GetBuffer()
43 inline const T* GetBuffer(const RunTimeOperandInfo* operand) { in GetBuffer()
48 inline const T* GetOptionalBuffer(const RunTimeOperandInfo* operand) { in GetOptionalBuffer()
82 RunTimeOperandInfo* operands) { in BidirectionalSequenceLSTM()
209 bool BidirectionalSequenceLSTM::Prepare(const Operation& operation, RunTimeOperandInfo* operands, in Prepare()
392 const RunTimeOperandInfo* bw_input = in Prepare()
DHashtableLookup.cpp40 HashtableLookup::HashtableLookup(const Operation& operation, RunTimeOperandInfo* operands) { in HashtableLookup()
/frameworks/ml/nn/common/include/
DCpuExecutor.h38 struct RunTimeOperandInfo { struct
163 std::vector<RunTimeOperandInfo> initializeRunTimeInfo(const hal::Subgraph& subgraph);
169 RunTimeOperandInfo* operands);
171 int executeSubgraph(const hal::Subgraph& subgraph, RunTimeOperandInfo* operands);
173 int executeOperation(const hal::Operation& operation, RunTimeOperandInfo* operands);
174 int executeIfOperation(const hal::Operation& operation, RunTimeOperandInfo* operands);
175 int executeWhileOperation(const hal::Operation& operation, RunTimeOperandInfo* operands);
178 const std::vector<RunTimeOperandInfo>& operands);
247 T getScalarData(const RunTimeOperandInfo& info) { in getScalarData()
254 T getScalarDataWithDefault(const RunTimeOperandInfo& info, T defaultValue) { in getScalarDataWithDefault()
[all …]
/frameworks/ml/nn/common/
DCpuExecutor.cpp55 OperationExecutionContext(const Operation* operation, RunTimeOperandInfo* operands) in OperationExecutionContext()
83 const RunTimeOperandInfo* getInputInfo(uint32_t index) const;
84 const RunTimeOperandInfo* getOutputInfo(uint32_t index) const;
85 RunTimeOperandInfo* getOutputInfo(uint32_t index);
88 RunTimeOperandInfo* operands;
93 const RunTimeOperandInfo* OperationExecutionContext::getInputInfo(uint32_t index) const { in getInputInfo()
98 const RunTimeOperandInfo* OperationExecutionContext::getOutputInfo(uint32_t index) const { in getOutputInfo()
103 RunTimeOperandInfo* OperationExecutionContext::getOutputInfo(uint32_t index) { in getOutputInfo()
154 bool setInfoAndAllocateIfNeeded(RunTimeOperandInfo* info, const Shape& shape, int* result) { in setInfoAndAllocateIfNeeded()
517 static bool convertToNhwc(RunTimeOperandInfo& to, const RunTimeOperandInfo& from, in convertToNhwc()
[all …]