Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DLSTM.h45 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 …]
DRNN.h35 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_;
DSVDF.h43 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_;
DHashtableLookup.h35 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_;
DLSHProjection.cpp27 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 …]
DLSHProjection.h41 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_;
DEmbeddingLookup.h35 struct RunTimeOperandInfo;
41 std::vector<RunTimeOperandInfo> &operands);
51 const RunTimeOperandInfo *value_;
52 const RunTimeOperandInfo *lookup_;
54 RunTimeOperandInfo *output_;
DLSTM.cpp28 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 …]
DSVDF.cpp28 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()
DRNN.cpp26 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()
DEmbeddingLookup.cpp27 std::vector<RunTimeOperandInfo>& operands) { in EmbeddingLookup()
DHashtableLookup.cpp35 std::vector<RunTimeOperandInfo>& operands) { in HashtableLookup()
/frameworks/ml/nn/common/
DCpuExecutor.cpp158 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/
DCpuExecutor.h33 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 …]