Home
last modified time | relevance | path

Searched refs:SizeOfDimension (Results 1 – 8 of 8) sorted by relevance

/packages/modules/NeuralNetworks/common/cpu_operations/
DLSHProjection.cpp57 NN_CHECK(SizeOfDimension(hash, 1) <= 32); in Prepare()
69 outputShape->dimensions = {SizeOfDimension(hash, 0)}; in Prepare()
75 NN_CHECK_EQ(SizeOfDimension(weight, 0), SizeOfDimension(input, 0)); in Prepare()
76 outputShape->dimensions = {SizeOfDimension(hash, 0) * SizeOfDimension(hash, 1)}; in Prepare()
99 SizeOfDimension(input, 0); in runningSignBit()
106 for (uint32_t i = 0; i < SizeOfDimension(input, 0); ++i) { in runningSignBit()
128 int num_hash = SizeOfDimension(hash, 0); in SparseLshProjection()
129 int num_bits = SizeOfDimension(hash, 1); in SparseLshProjection()
148 int num_hash = SizeOfDimension(hash, 0); in DenseLshProjection()
149 int num_bits = SizeOfDimension(hash, 1); in DenseLshProjection()
DBidirectionalSequenceLSTM.cpp268 const uint32_t max_time = SizeOfDimension(input_, params_.time_major ? 0 : 1); in Prepare()
269 const uint32_t n_batch = SizeOfDimension(input_, params_.time_major ? 1 : 0); in Prepare()
270 const uint32_t n_fw_input = SizeOfDimension(input_, 2); in Prepare()
272 const uint32_t n_fw_cell = SizeOfDimension(fw_input_to_output_weights_, 0); in Prepare()
274 NN_CHECK_EQ(SizeOfDimension(fw_input_to_output_weights_, 1), n_fw_input); in Prepare()
277 NN_CHECK_EQ(SizeOfDimension(fw_recurrent_to_output_weights_, 0), n_fw_cell); in Prepare()
278 const uint32_t n_fw_output = SizeOfDimension(fw_recurrent_to_output_weights_, 1); in Prepare()
280 const uint32_t n_bw_cell = SizeOfDimension(bw_input_to_output_weights_, 0); in Prepare()
283 NN_CHECK_EQ(SizeOfDimension(bw_recurrent_to_output_weights_, 0), n_bw_cell); in Prepare()
284 const uint32_t n_bw_output = SizeOfDimension(bw_recurrent_to_output_weights_, 1); in Prepare()
[all …]
DSVDF.cpp83 const uint32_t batch_size = SizeOfDimension(input, 0); in Prepare()
84 const uint32_t num_filters = SizeOfDimension(weights_feature, 0); in Prepare()
87 const uint32_t memory_size = SizeOfDimension(weights_time, 1); in Prepare()
88 NN_CHECK_EQ(SizeOfDimension(input, 1), SizeOfDimension(weights_feature, 1)); in Prepare()
89 NN_CHECK_EQ(SizeOfDimension(weights_time, 0), num_filters); in Prepare()
93 NN_CHECK_EQ(SizeOfDimension(bias, 0), num_units); in Prepare()
170 const int batch_size = SizeOfDimension(input_, 0); in EvalFloat32()
171 const int input_size = SizeOfDimension(input_, 1); in EvalFloat32()
172 const int num_filters = SizeOfDimension(weights_feature_, 0); in EvalFloat32()
174 const int memory_size = SizeOfDimension(weights_time_, 1); in EvalFloat32()
DQuantizedLSTM.cpp255 const uint32_t numBatches = SizeOfDimension(input, 0); in prepare()
256 const uint32_t inputSize = SizeOfDimension(input, 1); in prepare()
260 NN_RET_CHECK_EQ(SizeOfDimension(prevOutput, 0), numBatches); in prepare()
263 const uint32_t outputSize = SizeOfDimension(prevOutput, 1); in prepare()
272 NN_RET_CHECK_EQ(SizeOfDimension(weights, 0), outputSize); in prepare()
273 NN_RET_CHECK_EQ(SizeOfDimension(weights, 1), columns); in prepare()
304 NN_RET_CHECK_EQ(SizeOfDimension(bias, 0), outputSize); in prepare()
320 NN_CHECK_EQ(SizeOfDimension(prevCellState, 0), numBatches); in prepare()
321 NN_CHECK_EQ(SizeOfDimension(prevCellState, 1), outputSize); in prepare()
355 const int outputSize = SizeOfDimension(inputToInputWeights_, 0); in concatenateWeights()
[all …]
DLSTM.cpp160 NN_CHECK_EQ(SizeOfDimension(input_to_input_weights, 0), n_cell); in CheckInputTensorDimensions()
161 NN_CHECK_EQ(SizeOfDimension(input_to_input_weights, 1), n_input); in CheckInputTensorDimensions()
165 NN_CHECK_EQ(SizeOfDimension(input_to_forget_weights, 0), n_cell); in CheckInputTensorDimensions()
166 NN_CHECK_EQ(SizeOfDimension(input_to_forget_weights, 1), n_input); in CheckInputTensorDimensions()
169 NN_CHECK_EQ(SizeOfDimension(input_to_cell_weights, 0), n_cell); in CheckInputTensorDimensions()
170 NN_CHECK_EQ(SizeOfDimension(input_to_cell_weights, 1), n_input); in CheckInputTensorDimensions()
174 NN_CHECK_EQ(SizeOfDimension(recurrent_to_input_weights, 0), n_cell); in CheckInputTensorDimensions()
175 NN_CHECK_EQ(SizeOfDimension(recurrent_to_input_weights, 1), n_output); in CheckInputTensorDimensions()
179 NN_CHECK_EQ(SizeOfDimension(recurrent_to_forget_weights, 0), n_cell); in CheckInputTensorDimensions()
180 NN_CHECK_EQ(SizeOfDimension(recurrent_to_forget_weights, 1), n_output); in CheckInputTensorDimensions()
[all …]
DRNN.cpp61 const uint32_t batch_size = SizeOfDimension(input, 0); in Prepare()
62 const uint32_t num_units = SizeOfDimension(input_weights, 0); in Prepare()
63 NN_CHECK_EQ(SizeOfDimension(input, 1), SizeOfDimension(input_weights, 1)); in Prepare()
64 NN_CHECK_EQ(SizeOfDimension(input_weights, 0), SizeOfDimension(bias, 0)); in Prepare()
65 NN_CHECK_EQ(SizeOfDimension(recurrent_weights, 0), SizeOfDimension(bias, 0)); in Prepare()
66 NN_CHECK_EQ(SizeOfDimension(recurrent_weights, 1), SizeOfDimension(bias, 0)); in Prepare()
DMultinomial.cpp78 const uint32_t batch_size = SizeOfDimension(input, 0); in Prepare()
112 const uint32_t batch_size = SizeOfDimension(input_, 0); in EvalFloat32()
113 const uint32_t class_size = SizeOfDimension(input_, 1); in EvalFloat32()
/packages/modules/NeuralNetworks/common/include/
DCpuExecutor.h279 inline uint32_t SizeOfDimension(const RunTimeOperandInfo* operand, int i) { in SizeOfDimension() function