Searched refs:RunTimeOperandInfo (Results 1 – 14 of 14) sorted by relevance
/frameworks/ml/nn/common/operations/ |
D | LSTM.h | 45 struct RunTimeOperandInfo; 51 std::vector<RunTimeOperandInfo> &operands); 54 std::vector<RunTimeOperandInfo> &operands, 108 std::vector<RunTimeOperandInfo> &operands, uint32_t n_input, 112 const RunTimeOperandInfo *input_; 114 const RunTimeOperandInfo *input_to_input_weights_; 115 const RunTimeOperandInfo *input_to_forget_weights_; 116 const RunTimeOperandInfo *input_to_cell_weights_; 117 const RunTimeOperandInfo *input_to_output_weights_; 119 const RunTimeOperandInfo *recurrent_to_input_weights_; [all …]
|
D | RNN.h | 35 struct RunTimeOperandInfo; 41 std::vector<RunTimeOperandInfo> &operands); 44 std::vector<RunTimeOperandInfo> &operands, 62 const RunTimeOperandInfo *input_; 63 const RunTimeOperandInfo *weights_; 64 const RunTimeOperandInfo *recurrent_weights_; 65 const RunTimeOperandInfo *bias_; 66 const RunTimeOperandInfo *hidden_state_in_; 68 RunTimeOperandInfo *hidden_state_out_; 69 RunTimeOperandInfo *output_;
|
D | SVDF.h | 43 struct RunTimeOperandInfo; 49 std::vector<RunTimeOperandInfo>& operands); 53 std::vector<RunTimeOperandInfo> &operands, Shape *stateShape, 71 const RunTimeOperandInfo *input_; 72 const RunTimeOperandInfo *weights_feature_; 73 const RunTimeOperandInfo *weights_time_; 74 const RunTimeOperandInfo *bias_; 75 const RunTimeOperandInfo *state_in_; 77 RunTimeOperandInfo *state_out_; 78 RunTimeOperandInfo *output_;
|
D | HashtableLookup.h | 35 struct RunTimeOperandInfo; 41 std::vector<RunTimeOperandInfo> &operands); 53 const RunTimeOperandInfo *lookup_; 54 const RunTimeOperandInfo *key_; 55 const RunTimeOperandInfo *value_; 57 RunTimeOperandInfo *output_; 58 RunTimeOperandInfo *hits_;
|
D | LSHProjection.cpp | 27 std::vector<RunTimeOperandInfo>& operands) { in LSHProjection() 39 std::vector<RunTimeOperandInfo>& operands, in Prepare() 45 const RunTimeOperandInfo *hash = GetInput(operation, operands, kHashTensor); in Prepare() 50 const RunTimeOperandInfo* input = GetInput(operation, operands, kInputTensor); in Prepare() 61 RunTimeOperandInfo *weight = GetInput(operation, operands, kWeightTensor); in Prepare() 84 int running_sign_bit(const RunTimeOperandInfo* input, in running_sign_bit() 85 const RunTimeOperandInfo* weight, float seed) { in running_sign_bit() 113 void SparseLshProjection(const RunTimeOperandInfo* hash, in SparseLshProjection() 114 const RunTimeOperandInfo* input, in SparseLshProjection() 115 const RunTimeOperandInfo* weight, int32_t* out_buf) { in SparseLshProjection() [all …]
|
D | LSHProjection.h | 41 struct RunTimeOperandInfo; 48 std::vector<RunTimeOperandInfo> &operands); 52 std::vector<RunTimeOperandInfo>& operands, 67 const RunTimeOperandInfo *hash_; 68 const RunTimeOperandInfo *input_; 69 const RunTimeOperandInfo *weight_; 71 RunTimeOperandInfo *output_;
|
D | EmbeddingLookup.h | 35 struct RunTimeOperandInfo; 41 std::vector<RunTimeOperandInfo> &operands); 51 const RunTimeOperandInfo *value_; 52 const RunTimeOperandInfo *lookup_; 54 RunTimeOperandInfo *output_;
|
D | LSTM.cpp | 28 inline T *GetBuffer(RunTimeOperandInfo* operand) { in GetBuffer() 33 inline const T *GetBuffer(const RunTimeOperandInfo* operand) { in GetBuffer() 40 std::vector<RunTimeOperandInfo>& operands) { in LSTMCell() 82 const Operation &operation, std::vector<RunTimeOperandInfo> &operands, in CheckInputTensorDimensions() 97 const RunTimeOperandInfo *input_to_input_weights = in CheckInputTensorDimensions() 105 const RunTimeOperandInfo *input_to_forget_weights = in CheckInputTensorDimensions() 111 const RunTimeOperandInfo *input_to_cell_weights = in CheckInputTensorDimensions() 117 const RunTimeOperandInfo *recurrent_to_input_weights = in CheckInputTensorDimensions() 125 const RunTimeOperandInfo *recurrent_to_forget_weights = in CheckInputTensorDimensions() 131 const RunTimeOperandInfo *recurrent_to_cell_weights = in CheckInputTensorDimensions() [all …]
|
D | SVDF.cpp | 28 inline T *GetBuffer(RunTimeOperandInfo* operand) { in GetBuffer() 33 inline const T *GetBuffer(const RunTimeOperandInfo* operand) { in GetBuffer() 40 std::vector<RunTimeOperandInfo>& operands) { in SVDF() 56 std::vector<RunTimeOperandInfo> &operands, in Prepare() 65 const RunTimeOperandInfo *input = in Prepare() 67 const RunTimeOperandInfo *weights_feature = in Prepare() 69 const RunTimeOperandInfo *weights_time = in Prepare() 83 const RunTimeOperandInfo *bias = in Prepare()
|
D | RNN.cpp | 26 std::vector<RunTimeOperandInfo>& operands) { in RNN() 41 std::vector<RunTimeOperandInfo> &operands, in Prepare() 49 const RunTimeOperandInfo *input = in Prepare() 51 const RunTimeOperandInfo *input_weights = in Prepare() 53 const RunTimeOperandInfo *recurrent_weights = in Prepare() 55 const RunTimeOperandInfo *bias = in Prepare()
|
D | EmbeddingLookup.cpp | 27 std::vector<RunTimeOperandInfo>& operands) { in EmbeddingLookup()
|
D | HashtableLookup.cpp | 35 std::vector<RunTimeOperandInfo>& operands) { in HashtableLookup()
|
/frameworks/ml/nn/common/ |
D | CpuExecutor.cpp | 158 static bool setInfoAndAllocateIfNeeded(RunTimeOperandInfo* info, const Shape& shape) { in setInfoAndAllocateIfNeeded() 234 RunTimeOperandInfo& to = mOperands[i]; in initializeRunTimeInfo() 278 RunTimeOperandInfo& to = mOperands[operandIndex]; in initializeRunTimeInfo() 363 const RunTimeOperandInfo& in1 = mOperands[ins[0]]; in executeOperation() 364 const RunTimeOperandInfo& in2 = mOperands[ins[1]]; in executeOperation() 367 RunTimeOperandInfo& out = mOperands[outs[0]]; in executeOperation() 396 const RunTimeOperandInfo& in1 = mOperands[ins[0]]; in executeOperation() 397 const RunTimeOperandInfo& in2 = mOperands[ins[1]]; in executeOperation() 400 RunTimeOperandInfo& out = mOperands[outs[0]]; in executeOperation() 429 const RunTimeOperandInfo& input = mOperands[ins[0]]; in executeOperation() [all …]
|
/frameworks/ml/nn/common/include/ |
D | CpuExecutor.h | 33 struct RunTimeOperandInfo { struct 137 std::vector<RunTimeOperandInfo> mOperands; 180 T getScalarData(const RunTimeOperandInfo& info) { in getScalarData() 186 inline bool IsNullInput(const RunTimeOperandInfo *input) { in IsNullInput() 191 std::vector<RunTimeOperandInfo> &operands) { in NumInputsWithValues() 203 inline size_t NumDimensions(const RunTimeOperandInfo *operand) { in NumDimensions() 207 inline uint32_t SizeOfDimension(const RunTimeOperandInfo *operand, int i) { in SizeOfDimension() 211 inline RunTimeOperandInfo *GetInput(const Operation &operation, in GetInput() 212 std::vector<RunTimeOperandInfo> &operands, in GetInput() 217 inline RunTimeOperandInfo *GetOutput(const Operation &operation, in GetOutput() [all …]
|