Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DBidirectionalSequenceLSTM.h32 struct RunTimeOperandInfo;
37 std::vector<RunTimeOperandInfo>& operands);
39 bool Prepare(const Operation& operation, std::vector<RunTimeOperandInfo>& operands,
156 const RunTimeOperandInfo* input_;
158 const RunTimeOperandInfo* aux_input_;
159 const RunTimeOperandInfo* fw_aux_input_to_input_weights_;
160 const RunTimeOperandInfo* fw_aux_input_to_forget_weights_;
161 const RunTimeOperandInfo* fw_aux_input_to_cell_weights_;
162 const RunTimeOperandInfo* fw_aux_input_to_output_weights_;
163 const RunTimeOperandInfo* bw_aux_input_to_input_weights_;
[all …]
DLSTM.h43 struct RunTimeOperandInfo;
48 LSTMCell(const Operation& operation, std::vector<RunTimeOperandInfo>& operands);
50 bool Prepare(const Operation& operation, std::vector<RunTimeOperandInfo>& operands,
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.h12 struct RunTimeOperandInfo;
17 std::vector<RunTimeOperandInfo>& operands);
20 std::vector<RunTimeOperandInfo>& operands, Shape* cellStateShape,
52 const RunTimeOperandInfo* input_;
54 const RunTimeOperandInfo* inputToInputWeights_;
55 const RunTimeOperandInfo* inputToForgetWeights_;
56 const RunTimeOperandInfo* inputToCellWeights_;
57 const RunTimeOperandInfo* inputToOutputWeights_;
59 const RunTimeOperandInfo* recurrentToInputWeights_;
60 const RunTimeOperandInfo* recurrentToForgetWeights_;
[all …]
DLSHProjection.h34 struct RunTimeOperandInfo;
39 LSHProjection(const Operation& operation, std::vector<RunTimeOperandInfo>& operands);
41 static bool Prepare(const Operation& operation, std::vector<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.cpp29 std::vector<RunTimeOperandInfo>& operands) { in LSHProjection()
40 bool LSHProjection::Prepare(const Operation& operation, std::vector<RunTimeOperandInfo>& operands, in Prepare()
46 const RunTimeOperandInfo* hash = GetInput(operation, operands, kHashTensor); in Prepare()
51 const RunTimeOperandInfo* input = GetInput(operation, operands, kInputTensor); in Prepare()
63 RunTimeOperandInfo* weight = GetInput(operation, operands, kWeightTensor); in Prepare()
87 int runningSignBit(const RunTimeOperandInfo* input, const RunTimeOperandInfo* weight, float seed) { in runningSignBit()
116 void SparseLshProjection(LSHProjectionType type, const RunTimeOperandInfo* hash, in SparseLshProjection()
117 const RunTimeOperandInfo* input, const RunTimeOperandInfo* weight, in SparseLshProjection()
137 void DenseLshProjection(const RunTimeOperandInfo* hash, const RunTimeOperandInfo* input, in DenseLshProjection()
138 const RunTimeOperandInfo* weight, int32_t* out_buf) { in DenseLshProjection()
[all …]
DSVDF.h34 struct RunTimeOperandInfo;
39 SVDF(const Operation& operation, std::vector<RunTimeOperandInfo>& operands);
41 static bool Prepare(const Operation& operation, std::vector<RunTimeOperandInfo>& operands,
63 const RunTimeOperandInfo* input_;
64 const RunTimeOperandInfo* weights_feature_;
65 const RunTimeOperandInfo* weights_time_;
66 const RunTimeOperandInfo* bias_;
67 const RunTimeOperandInfo* state_in_;
69 RunTimeOperandInfo* state_out_;
70 RunTimeOperandInfo* output_;
DRNN.h26 struct RunTimeOperandInfo;
31 RNN(const Operation& operation, std::vector<RunTimeOperandInfo>& operands);
33 static bool Prepare(const Operation& operation, std::vector<RunTimeOperandInfo>& operands,
65 const RunTimeOperandInfo* input_;
66 const RunTimeOperandInfo* weights_;
67 const RunTimeOperandInfo* recurrent_weights_;
68 const RunTimeOperandInfo* bias_;
69 const RunTimeOperandInfo* hidden_state_in_;
71 RunTimeOperandInfo* hidden_state_out_;
72 RunTimeOperandInfo* output_;
DHashtableLookup.h27 struct RunTimeOperandInfo;
33 std::vector<RunTimeOperandInfo> &operands);
45 const RunTimeOperandInfo *lookup_;
46 const RunTimeOperandInfo *key_;
47 const RunTimeOperandInfo *value_;
49 RunTimeOperandInfo *output_;
50 RunTimeOperandInfo *hits_;
DMultinomial.h30 struct RunTimeOperandInfo;
36 std::vector<RunTimeOperandInfo>& operands);
39 std::vector<RunTimeOperandInfo>& operands, Shape* outputShape);
51 RunTimeOperandInfo* input_;
53 RunTimeOperandInfo* random_seeds_;
55 RunTimeOperandInfo* output_;
DEmbeddingLookup.h27 struct RunTimeOperandInfo;
33 std::vector<RunTimeOperandInfo> &operands);
43 const RunTimeOperandInfo *value_;
44 const RunTimeOperandInfo *lookup_;
46 RunTimeOperandInfo *output_;
DMultinomial.cpp36 inline T* GetBuffer(RunTimeOperandInfo* operand) { in GetBuffer()
41 inline const T* GetBuffer(const RunTimeOperandInfo* operand) { in GetBuffer()
47 Multinomial::Multinomial(const Operation& operation, std::vector<RunTimeOperandInfo>& operands) { in Multinomial()
56 bool Multinomial::Prepare(const Operation& operation, std::vector<RunTimeOperandInfo>& operands, in Prepare()
62 const RunTimeOperandInfo* input = GetInput(operation, operands, Multinomial::kInputTensor); in Prepare()
DLSTM.cpp33 inline T* GetBuffer(RunTimeOperandInfo* operand) { in GetBuffer()
38 inline const T* GetBuffer(const RunTimeOperandInfo* operand) { in GetBuffer()
43 inline const T* GetOptionalBuffer(const RunTimeOperandInfo* operand) { in GetOptionalBuffer()
49 LSTMCell::LSTMCell(const Operation& operation, std::vector<RunTimeOperandInfo>& operands) { in LSTMCell()
106 static RunTimeOperandInfo no_value; in LSTMCell()
124 const RunTimeOperandInfo* input_, const RunTimeOperandInfo* input_to_input_weights, in CheckInputTensorDimensions()
125 const RunTimeOperandInfo* input_to_forget_weights, in CheckInputTensorDimensions()
126 const RunTimeOperandInfo* input_to_cell_weights, in CheckInputTensorDimensions()
127 const RunTimeOperandInfo* input_to_output_weights, in CheckInputTensorDimensions()
128 const RunTimeOperandInfo* recurrent_to_input_weights, in CheckInputTensorDimensions()
[all …]
DRNN.cpp29 std::vector<RunTimeOperandInfo>& operands) { in RNN()
45 std::vector<RunTimeOperandInfo> &operands, in Prepare()
54 const RunTimeOperandInfo *input = in Prepare()
56 const RunTimeOperandInfo *input_weights = in Prepare()
58 const RunTimeOperandInfo *recurrent_weights = in Prepare()
60 const RunTimeOperandInfo *bias = in Prepare()
DSVDF.cpp29 std::vector<RunTimeOperandInfo>& operands) { in SVDF()
46 std::vector<RunTimeOperandInfo> &operands, in Prepare()
56 const RunTimeOperandInfo *input = in Prepare()
58 const RunTimeOperandInfo *weights_feature = in Prepare()
60 const RunTimeOperandInfo *weights_time = in Prepare()
74 const RunTimeOperandInfo *bias = in Prepare()
DQuantizedLSTM.cpp33 inline T* GetBuffer(RunTimeOperandInfo* operand) { in GetBuffer()
38 inline const T* GetBuffer(const RunTimeOperandInfo* operand) { in GetBuffer()
202 void assignWeightsSubmatrix(const RunTimeOperandInfo* submatrix, const int32_t offset_row, in assignWeightsSubmatrix()
217 std::vector<RunTimeOperandInfo>& operands) { in QuantizedLSTMCell()
243 std::vector<RunTimeOperandInfo>& operands, Shape* cellStateOutShape, in prepare()
264 auto checkWeightsShape = [&](const RunTimeOperandInfo* weights, uint32_t columns) -> bool { in prepare()
296 auto checkBiasShape = [&](const RunTimeOperandInfo* bias) -> bool { in prepare()
DEmbeddingLookup.cpp28 std::vector<RunTimeOperandInfo>& operands) { in EmbeddingLookup()
DHashtableLookup.cpp36 std::vector<RunTimeOperandInfo>& operands) { in HashtableLookup()
DBidirectionalSequenceLSTM.cpp32 inline T* GetBuffer(RunTimeOperandInfo* operand) { in GetBuffer()
37 inline const T* GetBuffer(const RunTimeOperandInfo* operand) { in GetBuffer()
42 inline const T* GetOptionalBuffer(const RunTimeOperandInfo* operand) { in GetOptionalBuffer()
49 std::vector<RunTimeOperandInfo>& operands) { in BidirectionalSequenceLSTM()
161 std::vector<RunTimeOperandInfo>& operands, in Prepare()
/frameworks/ml/nn/common/include/
DCpuExecutor.h36 struct RunTimeOperandInfo { struct
176 std::vector<RunTimeOperandInfo> mOperands;
230 T getScalarData(const RunTimeOperandInfo& info) { in getScalarData()
236 inline bool IsNullInput(const RunTimeOperandInfo *input) { in IsNullInput()
241 std::vector<RunTimeOperandInfo> &operands) { in NumInputsWithValues()
253 inline size_t NumDimensions(const RunTimeOperandInfo *operand) { in NumDimensions()
257 inline uint32_t SizeOfDimension(const RunTimeOperandInfo *operand, int i) { in SizeOfDimension()
261 inline RunTimeOperandInfo *GetInput(const Operation &operation, in GetInput()
262 std::vector<RunTimeOperandInfo> &operands, in GetInput()
267 inline RunTimeOperandInfo *GetOutput(const Operation &operation, in GetOutput()
[all …]
/frameworks/ml/nn/common/
DCpuExecutor.cpp44 OperationExecutionContext(const Operation* operation, RunTimeOperandInfo* operands) in OperationExecutionContext()
72 const RunTimeOperandInfo* getInputInfo(uint32_t index) const;
73 const RunTimeOperandInfo* getOutputInfo(uint32_t index) const;
74 RunTimeOperandInfo* getOutputInfo(uint32_t index);
77 RunTimeOperandInfo* operands;
82 const RunTimeOperandInfo* OperationExecutionContext::getInputInfo(uint32_t index) const { in getInputInfo()
87 const RunTimeOperandInfo* OperationExecutionContext::getOutputInfo(uint32_t index) const { in getOutputInfo()
92 RunTimeOperandInfo* OperationExecutionContext::getOutputInfo(uint32_t index) { in getOutputInfo()
140 bool setInfoAndAllocateIfNeeded(RunTimeOperandInfo* info, const Shape& shape, int* result) { in setInfoAndAllocateIfNeeded()
450 static bool convertToNhwc(RunTimeOperandInfo& to, const RunTimeOperandInfo& from, in convertToNhwc()
[all …]