Home
last modified time | relevance | path

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

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DBidirectionalSequenceLSTM.h31 struct RunTimeOperandInfo;
160 const RunTimeOperandInfo* input_;
162 const RunTimeOperandInfo* aux_input_;
188 const RunTimeOperandInfo* fw_cell_bias_;
200 const RunTimeOperandInfo* fw_cell_state_;
201 RunTimeOperandInfo* fw_output_;
219 const RunTimeOperandInfo* bw_cell_bias_;
231 const RunTimeOperandInfo* bw_cell_state_;
232 RunTimeOperandInfo* bw_output_;
235 RunTimeOperandInfo* fw_output_cell_state_;
[all …]
H A DLSTM.h46 struct RunTimeOperandInfo;
185 const RunTimeOperandInfo* input_, const RunTimeOperandInfo* input_to_input_weights,
196 const RunTimeOperandInfo* input_gate_bias, const RunTimeOperandInfo* forget_gate_bias,
197 const RunTimeOperandInfo* cell_bias, const RunTimeOperandInfo* output_gate_bias,
198 const RunTimeOperandInfo* projection_weights, const RunTimeOperandInfo* projection_bias,
207 const RunTimeOperandInfo* input_;
225 const RunTimeOperandInfo* cell_bias_;
239 RunTimeOperandInfo* output_state_out_;
240 RunTimeOperandInfo* cell_state_out_;
241 RunTimeOperandInfo* output_;
[all …]
H A DQuantizedLSTM.h27 struct RunTimeOperandInfo;
65 const RunTimeOperandInfo* input_;
69 const RunTimeOperandInfo* inputToCellWeights_;
77 const RunTimeOperandInfo* inputGateBias_;
78 const RunTimeOperandInfo* forgetGateBias_;
79 const RunTimeOperandInfo* cellGateBias_;
80 const RunTimeOperandInfo* outputGateBias_;
82 const RunTimeOperandInfo* prevCellState_;
83 const RunTimeOperandInfo* prevOutput_;
85 RunTimeOperandInfo* cellStateOut_;
[all …]
H A DLSHProjection.cpp96 int runningSignBit(const RunTimeOperandInfo* input, const RunTimeOperandInfo* weight, float seed) { in runningSignBit()
126 const RunTimeOperandInfo* input, const RunTimeOperandInfo* weight, in SparseLshProjection()
146 void DenseLshProjection(const RunTimeOperandInfo* hash, const RunTimeOperandInfo* input, in DenseLshProjection()
182 template int runningSignBit<float>(const RunTimeOperandInfo* input,
184 template int runningSignBit<_Float16>(const RunTimeOperandInfo* input,
188 const RunTimeOperandInfo* input,
191 const RunTimeOperandInfo* input,
194 template void DenseLshProjection<float>(const RunTimeOperandInfo* hash,
195 const RunTimeOperandInfo* input,
197 template void DenseLshProjection<_Float16>(const RunTimeOperandInfo* hash,
[all …]
H A 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);
69 const RunTimeOperandInfo* input, const RunTimeOperandInfo* weight,
73 void DenseLshProjection(const RunTimeOperandInfo* hash, const RunTimeOperandInfo* input,
[all …]
H A 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_;
H A 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_;
H A 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_;
H A 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_;
H A DEmbeddingLookup.h27 struct RunTimeOperandInfo;
31 EmbeddingLookup(const Operation& operation, RunTimeOperandInfo* operands);
41 const RunTimeOperandInfo* value_;
42 const RunTimeOperandInfo* lookup_;
44 RunTimeOperandInfo* output_;
H A DLSTM.cpp38 inline T* GetBuffer(RunTimeOperandInfo* operand) { in GetBuffer()
43 inline const T* GetBuffer(const RunTimeOperandInfo* operand) { in GetBuffer()
115 static RunTimeOperandInfo no_value; in LSTMCell()
133 const RunTimeOperandInfo* input_, const RunTimeOperandInfo* input_to_input_weights, in CheckInputTensorDimensions()
135 const RunTimeOperandInfo* input_to_cell_weights, in CheckInputTensorDimensions()
141 const RunTimeOperandInfo* cell_to_input_weights, in CheckInputTensorDimensions()
142 const RunTimeOperandInfo* cell_to_forget_weights, in CheckInputTensorDimensions()
143 const RunTimeOperandInfo* cell_to_output_weights, const RunTimeOperandInfo* input_gate_bias, in CheckInputTensorDimensions()
144 const RunTimeOperandInfo* forget_gate_bias, const RunTimeOperandInfo* cell_bias, in CheckInputTensorDimensions()
145 const RunTimeOperandInfo* output_gate_bias, const RunTimeOperandInfo* projection_weights, in CheckInputTensorDimensions()
[all …]
H A 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()
H A 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()
H A 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()
H A 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()
H A 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()
H A DEmbeddingLookup.cpp28 EmbeddingLookup::EmbeddingLookup(const Operation& operation, RunTimeOperandInfo* operands) { in EmbeddingLookup()
H A DHashtableLookup.cpp36 HashtableLookup::HashtableLookup(const Operation& operation, RunTimeOperandInfo* operands) { in HashtableLookup()
/aosp12/packages/modules/NeuralNetworks/common/include/
H A DCpuExecutor.h38 struct RunTimeOperandInfo { struct
169 RunTimeOperandInfo* operands);
173 int executeOperation(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()
261 inline bool IsNullInput(const RunTimeOperandInfo* input) { in IsNullInput()
275 inline size_t NumDimensions(const RunTimeOperandInfo* operand) { in NumDimensions()
279 inline uint32_t SizeOfDimension(const RunTimeOperandInfo* operand, int i) { in SizeOfDimension()
283 inline RunTimeOperandInfo* GetInput(const Operation& operation, RunTimeOperandInfo* operands, in GetInput()
[all …]
/aosp12/packages/modules/NeuralNetworks/common/
H A DCpuExecutor.cpp99 RunTimeOperandInfo* getOutputInfo(uint32_t index);
102 RunTimeOperandInfo* operands;
434 static bool convertToNhwc(RunTimeOperandInfo& to, const RunTimeOperandInfo& from, in convertToNhwc()
478 static bool convertFromNhwc(RunTimeOperandInfo& to, const RunTimeOperandInfo& from, in convertFromNhwc()
613 std::vector<RunTimeOperandInfo> operands(count); in initializeRunTimeInfo()
618 RunTimeOperandInfo& to = operands[i]; in initializeRunTimeInfo()
822 RunTimeOperandInfo input_tmp, output_tmp; in executeOperation()
1001 RunTimeOperandInfo& fwOutput = in executeOperation()
1012 RunTimeOperandInfo& bwOutput = in executeOperation()
1619 RunTimeOperandInfo& cellStateOut = in executeOperation()
[all …]
/aosp12/packages/modules/NeuralNetworks/driver/sample/
H A 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()
763 const RunTimeOperandInfo& filter = operands[ins[1]]; in VisitConv2DNode()
862 const RunTimeOperandInfo& filter = operands[ins[1]]; in VisitDepthwiseConv2DNode()
[all …]