Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/operations/
DBidirectionalSequenceLSTM.h31 struct RunTimeOperandInfo;
35 BidirectionalSequenceLSTM(const Operation& operation, RunTimeOperandInfo* operands);
37 bool Prepare(const Operation& operation, RunTimeOperandInfo* operands, Shape* fwOutputShape,
160 const RunTimeOperandInfo* input_;
162 const RunTimeOperandInfo* aux_input_;
163 const RunTimeOperandInfo* fw_aux_input_to_input_weights_;
164 const RunTimeOperandInfo* fw_aux_input_to_forget_weights_;
165 const RunTimeOperandInfo* fw_aux_input_to_cell_weights_;
166 const RunTimeOperandInfo* fw_aux_input_to_output_weights_;
167 const RunTimeOperandInfo* bw_aux_input_to_input_weights_;
[all …]
DLSTM.h46 struct RunTimeOperandInfo;
51 LSTMCell(const Operation& operation, RunTimeOperandInfo* operands);
53 bool Prepare(const 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 …]
DLSHProjection.cpp33 LSHProjection::LSHProjection(const Operation& operation, RunTimeOperandInfo* operands) { in LSHProjection()
44 bool LSHProjection::Prepare(const Operation& operation, RunTimeOperandInfo* operands, in Prepare()
54 const RunTimeOperandInfo* hash = GetInput(operation, operands, kHashTensor); in Prepare()
59 const RunTimeOperandInfo* input = GetInput(operation, operands, kInputTensor); in Prepare()
72 RunTimeOperandInfo* weight = GetInput(operation, operands, kWeightTensor); in Prepare()
96 int runningSignBit(const RunTimeOperandInfo* input, const RunTimeOperandInfo* weight, float seed) { in runningSignBit()
125 void SparseLshProjection(LSHProjectionType type, const RunTimeOperandInfo* hash, in SparseLshProjection()
126 const RunTimeOperandInfo* input, const RunTimeOperandInfo* weight, in SparseLshProjection()
146 void DenseLshProjection(const RunTimeOperandInfo* hash, const RunTimeOperandInfo* input, in DenseLshProjection()
147 const RunTimeOperandInfo* weight, int32_t* out_buf) { in DenseLshProjection()
[all …]
DQuantizedLSTM.h27 struct RunTimeOperandInfo;
31 QuantizedLSTMCell(const Operation& operation, RunTimeOperandInfo* operands);
33 static bool prepare(const 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 Operation& operation, RunTimeOperandInfo* operands);
41 static bool Prepare(const 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 …]
DSVDF.h36 struct RunTimeOperandInfo;
41 SVDF(const Operation& operation, RunTimeOperandInfo* operands);
43 static bool Prepare(const Operation& operation, RunTimeOperandInfo* operands, Shape* stateShape,
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 Operation& operation, RunTimeOperandInfo* operands);
35 static bool Prepare(const 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 Operation& operation, RunTimeOperandInfo* operands);
43 const RunTimeOperandInfo* lookup_;
44 const RunTimeOperandInfo* key_;
45 const RunTimeOperandInfo* value_;
47 RunTimeOperandInfo* output_;
48 RunTimeOperandInfo* hits_;
DMultinomial.h29 struct RunTimeOperandInfo;
34 Multinomial(const Operation& operation, RunTimeOperandInfo* operands);
36 static bool Prepare(const Operation& operation, RunTimeOperandInfo* operands,
49 RunTimeOperandInfo* input_;
51 RunTimeOperandInfo* random_seeds_;
53 RunTimeOperandInfo* output_;
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 …]
DEmbeddingLookup.h27 struct RunTimeOperandInfo;
31 EmbeddingLookup(const Operation& operation, RunTimeOperandInfo* operands);
41 const RunTimeOperandInfo* value_;
42 const RunTimeOperandInfo* lookup_;
44 RunTimeOperandInfo* output_;
DMultinomial.cpp45 inline T* GetBuffer(RunTimeOperandInfo* operand) { in GetBuffer()
50 inline const T* GetBuffer(const RunTimeOperandInfo* operand) { in GetBuffer()
56 Multinomial::Multinomial(const Operation& operation, RunTimeOperandInfo* operands) { in Multinomial()
65 bool Multinomial::Prepare(const Operation& operation, RunTimeOperandInfo* operands, in Prepare()
71 const RunTimeOperandInfo* input = GetInput(operation, operands, Multinomial::kInputTensor); in Prepare()
DSVDF.cpp31 SVDF::SVDF(const Operation& operation, RunTimeOperandInfo* operands) { in SVDF()
49 bool SVDF::Prepare(const Operation& operation, RunTimeOperandInfo* operands, Shape* stateShape, in Prepare()
72 const RunTimeOperandInfo* input = GetInput(operation, operands, SVDF::kInputTensor); in Prepare()
73 const RunTimeOperandInfo* weights_feature = in Prepare()
75 const RunTimeOperandInfo* weights_time = in Prepare()
89 const RunTimeOperandInfo* bias = GetInput(operation, operands, kBiasTensor); in Prepare()
DRNN.cpp30 RNN::RNN(const Operation& operation, RunTimeOperandInfo* operands) { in RNN()
45 bool RNN::Prepare(const Operation& operation, RunTimeOperandInfo* operands, Shape* hiddenStateShape, in Prepare()
53 const RunTimeOperandInfo* input = GetInput(operation, operands, kInputTensor); in Prepare()
54 const RunTimeOperandInfo* input_weights = GetInput(operation, operands, kWeightsTensor); in Prepare()
55 const RunTimeOperandInfo* recurrent_weights = in Prepare()
57 const RunTimeOperandInfo* bias = GetInput(operation, operands, kBiasTensor); in Prepare()
DQuantizedLSTM.cpp37 inline T* GetBuffer(RunTimeOperandInfo* operand) { in GetBuffer()
42 inline const T* GetBuffer(const RunTimeOperandInfo* operand) { in GetBuffer()
206 void assignWeightsSubmatrix(const RunTimeOperandInfo* submatrix, const int32_t offset_row, in assignWeightsSubmatrix()
220 QuantizedLSTMCell::QuantizedLSTMCell(const Operation& operation, RunTimeOperandInfo* operands) { in QuantizedLSTMCell()
245 bool QuantizedLSTMCell::prepare(const Operation& operation, RunTimeOperandInfo* operands, in prepare()
266 auto checkWeightsShape = [&](const RunTimeOperandInfo* weights, uint32_t columns) -> bool { in prepare()
298 auto checkBiasShape = [&](const RunTimeOperandInfo* bias) -> bool { in prepare()
DBidirectionalSequenceLSTM.cpp37 inline T* GetBuffer(RunTimeOperandInfo* operand) { in GetBuffer()
42 inline const T* GetBuffer(const RunTimeOperandInfo* operand) { in GetBuffer()
47 inline const T* GetOptionalBuffer(const RunTimeOperandInfo* operand) { in GetOptionalBuffer()
81 RunTimeOperandInfo* operands) { in BidirectionalSequenceLSTM()
208 bool BidirectionalSequenceLSTM::Prepare(const Operation& operation, RunTimeOperandInfo* operands, in Prepare()
391 const RunTimeOperandInfo* bw_input = in Prepare()
DEmbeddingLookup.cpp28 EmbeddingLookup::EmbeddingLookup(const Operation& operation, RunTimeOperandInfo* operands) { in EmbeddingLookup()
DHashtableLookup.cpp36 HashtableLookup::HashtableLookup(const Operation& operation, RunTimeOperandInfo* operands) { in HashtableLookup()
/packages/modules/NeuralNetworks/common/
DCpuExecutor.cpp69 OperationExecutionContext(const Operation* operation, RunTimeOperandInfo* operands) in OperationExecutionContext()
97 const RunTimeOperandInfo* getInputInfo(uint32_t index) const;
98 const RunTimeOperandInfo* getOutputInfo(uint32_t index) const;
99 RunTimeOperandInfo* getOutputInfo(uint32_t index);
102 RunTimeOperandInfo* operands;
107 const RunTimeOperandInfo* OperationExecutionContext::getInputInfo(uint32_t index) const { in getInputInfo()
112 const RunTimeOperandInfo* OperationExecutionContext::getOutputInfo(uint32_t index) const { in getOutputInfo()
117 RunTimeOperandInfo* OperationExecutionContext::getOutputInfo(uint32_t index) { in getOutputInfo()
168 bool setInfoAndAllocateIfNeeded(RunTimeOperandInfo* info, const Shape& shape, int* result) { in setInfoAndAllocateIfNeeded()
434 static bool convertToNhwc(RunTimeOperandInfo& to, const RunTimeOperandInfo& from, in convertToNhwc()
[all …]
/packages/modules/NeuralNetworks/common/include/
DCpuExecutor.h38 struct RunTimeOperandInfo { struct
163 std::vector<RunTimeOperandInfo> initializeRunTimeInfo(const Model::Subgraph& subgraph);
169 RunTimeOperandInfo* operands);
171 int executeSubgraph(const Model::Subgraph& subgraph, RunTimeOperandInfo* operands);
173 int executeOperation(const Operation& operation, RunTimeOperandInfo* operands);
174 int executeIfOperation(const Operation& operation, RunTimeOperandInfo* operands);
175 int executeWhileOperation(const 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 …]
/packages/modules/NeuralNetworks/driver/sample/
DSampleDriverFloatXNNPACK.cpp75 RunTimeOperandInfo* operands) { in updateForArguments()
80 RunTimeOperandInfo& to = operands[operandIndex]; in updateForArguments()
108 std::vector<RunTimeOperandInfo> initializeRunTimeInfo( in initializeRunTimeInfo()
112 std::vector<RunTimeOperandInfo> operands(count); in initializeRunTimeInfo()
115 RunTimeOperandInfo& to = operands[i]; in initializeRunTimeInfo()
157 std::vector<RunTimeOperandInfo>& operands, in Create()
273 V1_3::ErrorStatus Invoke(RunTimeOperandInfo* operands) { in Invoke()
497 RunTimeOperandInfo* operands, in VisitNode()
561 RunTimeOperandInfo* operands, in VisitAbsNode()
581 RunTimeOperandInfo* operands, in VisitAddNode()
[all …]