Lines Matching refs:dims
149 std::vector<uint32_t> dims = getAlignedDimensions(operand.dimensions, 4); in addOperand() local
150 HEXAGON_SOFT_ASSERT_NE(0ul, dims.size(), "Rank must be at most 4"); in addOperand()
152 createTensorInternal(dims[0], dims[1], dims[2], dims[3], operand.buffer, operand.length); in addOperand()
202 std::vector<uint32_t> dims = getAlignedDimensions(mOperands[operand].dimensions, 4); in createConvFilterTensor() local
203 HEXAGON_SOFT_ASSERT_NE(0ul, dims.size(), "Need at most 4 dimensions"); in createConvFilterTensor()
207 transpose<float>(dims[0], dims[1] * dims[2] * dims[3], in createConvFilterTensor()
209 return createTensorInternal(dims[1], dims[2], dims[3], dims[0], in createConvFilterTensor()
214 transpose<uint8_t>(dims[0], dims[1] * dims[2] * dims[3], in createConvFilterTensor()
216 return createTensorInternal(dims[1], dims[2], dims[3], dims[0], in createConvFilterTensor()
224 std::vector<uint32_t> dims = getAlignedDimensions(mOperands[operand].dimensions, 4); in createDepthwiseFilterTensor() local
225 HEXAGON_SOFT_ASSERT_NE(0ul, dims.size(), "Need at most 4 dimensions"); in createDepthwiseFilterTensor()
227 return createTensorInternal(dims[1], dims[2], dims[3] / depth_multiplier, in createDepthwiseFilterTensor()
228 dims[0] * depth_multiplier, operandInfo.buffer, operandInfo.length); in createDepthwiseFilterTensor()
233 std::vector<uint32_t> dims = getAlignedDimensions(mOperands[operand].dimensions, 4); in createFullyConnectedWeightTensor() local
234 HEXAGON_SOFT_ASSERT_NE(0ul, dims.size(), "Need at most 4 dimensions"); in createFullyConnectedWeightTensor()
236 uint32_t num_units = dims[0] * dims[1] * dims[2]; in createFullyConnectedWeightTensor()
237 uint32_t input_size = dims[3]; in createFullyConnectedWeightTensor()