Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DLSHProjection.cpp58 NN_CHECK(SizeOfDimension(hash, 1) <= 32); in Prepare()
70 outputShape->dimensions = {SizeOfDimension(hash, 0)}; in Prepare()
76 NN_CHECK_EQ(SizeOfDimension(weight, 0), SizeOfDimension(input, 0)); in Prepare()
77 outputShape->dimensions = {SizeOfDimension(hash, 0) * SizeOfDimension(hash, 1)}; in Prepare()
100 SizeOfDimension(input, 0); in runningSignBit()
107 for (uint32_t i = 0; i < SizeOfDimension(input, 0); ++i) { in runningSignBit()
129 int num_hash = SizeOfDimension(hash, 0); in SparseLshProjection()
130 int num_bits = SizeOfDimension(hash, 1); in SparseLshProjection()
149 int num_hash = SizeOfDimension(hash, 0); in DenseLshProjection()
150 int num_bits = SizeOfDimension(hash, 1); in DenseLshProjection()
DBidirectionalSequenceLSTM.cpp269 const uint32_t max_time = SizeOfDimension(input_, params_.time_major ? 0 : 1); in Prepare()
270 const uint32_t n_batch = SizeOfDimension(input_, params_.time_major ? 1 : 0); in Prepare()
271 const uint32_t n_fw_input = SizeOfDimension(input_, 2); in Prepare()
273 const uint32_t n_fw_cell = SizeOfDimension(fw_input_to_output_weights_, 0); in Prepare()
275 NN_CHECK_EQ(SizeOfDimension(fw_input_to_output_weights_, 1), n_fw_input); in Prepare()
278 NN_CHECK_EQ(SizeOfDimension(fw_recurrent_to_output_weights_, 0), n_fw_cell); in Prepare()
279 const uint32_t n_fw_output = SizeOfDimension(fw_recurrent_to_output_weights_, 1); in Prepare()
281 const uint32_t n_bw_cell = SizeOfDimension(bw_input_to_output_weights_, 0); in Prepare()
284 NN_CHECK_EQ(SizeOfDimension(bw_recurrent_to_output_weights_, 0), n_bw_cell); in Prepare()
285 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.cpp254 const uint32_t numBatches = SizeOfDimension(input, 0); in prepare()
255 const uint32_t inputSize = SizeOfDimension(input, 1); in prepare()
259 NN_RET_CHECK_EQ(SizeOfDimension(prevOutput, 0), numBatches); in prepare()
262 const uint32_t outputSize = SizeOfDimension(prevOutput, 1); in prepare()
271 NN_RET_CHECK_EQ(SizeOfDimension(weights, 0), outputSize); in prepare()
272 NN_RET_CHECK_EQ(SizeOfDimension(weights, 1), columns); in prepare()
303 NN_RET_CHECK_EQ(SizeOfDimension(bias, 0), outputSize); in prepare()
319 NN_CHECK_EQ(SizeOfDimension(prevCellState, 0), numBatches); in prepare()
320 NN_CHECK_EQ(SizeOfDimension(prevCellState, 1), outputSize); in prepare()
354 const int outputSize = SizeOfDimension(inputToInputWeights_, 0); in concatenateWeights()
[all …]
DRNN.cpp65 const uint32_t batch_size = SizeOfDimension(input, 0); in Prepare()
66 const uint32_t num_units = SizeOfDimension(input_weights, 0); in Prepare()
67 NN_CHECK_EQ(SizeOfDimension(input, 1), SizeOfDimension(input_weights, 1)); in Prepare()
68 NN_CHECK_EQ(SizeOfDimension(input_weights, 0), SizeOfDimension(bias, 0)); in Prepare()
69 NN_CHECK_EQ(SizeOfDimension(recurrent_weights, 0), SizeOfDimension(bias, 0)); in Prepare()
70 NN_CHECK_EQ(SizeOfDimension(recurrent_weights, 1), SizeOfDimension(bias, 0)); in Prepare()
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 …]
DMultinomial.cpp72 const uint32_t batch_size = SizeOfDimension(input, 0); in Prepare()
106 const int batch_size = SizeOfDimension(input_, 0); in EvalFloat32()
107 const int class_size = SizeOfDimension(input_, 1); in EvalFloat32()
/frameworks/ml/nn/common/include/
DCpuExecutor.h280 inline uint32_t SizeOfDimension(const RunTimeOperandInfo* operand, int i) { in SizeOfDimension() function