Searched defs:numInputElements (Results 1 – 8 of 8) sorted by relevance
32 const int64_t numInputElements, in max_unpooling2d_forward_kernel()76 const int64_t numInputElements, in max_unpooling2d_backward_kernel()
116 auto numInputElements = std::accumulate(std::begin(inputShape), in VerifyTosaAttribute() local
80 …OpAtomicBuffer(const uint32_t numInputElements, const uint32_t numOuptutElements, const OpAtomicTy… in OpAtomicBuffer()
141 RandomVariable numInputElements = 1; in reshapeConstructor() local
298 int32_t numInputElements = (int32_t)getNumberOfElements(input); in reshapePrepare() local
2247 unsigned int numInputElements = m_TensorsInfo[node.input(0)].m_info->GetNumElements(); in ParseReshape() local
1844 const unsigned int numInputElements = inputShape[0] * inputHeight * inputWidth * in Validate() local
3066 const int32_t numInputElements = inputTensorInfo.GetNumElements(); in ParseReshape() local