Searched refs:axisSize (Results 1 – 10 of 10) sorted by relevance
47 const unsigned int axisSize = inputShape[uAxis]; in LogSoftmax() local57 input[outer * axisSize * innerSize + inner]; in LogSoftmax()59 for (unsigned int i = 1u; i < axisSize; ++i) in LogSoftmax()61 input[(outer * axisSize + i) * innerSize + inner]; in LogSoftmax()67 for (unsigned int i = 0u; i < axisSize; ++i) in LogSoftmax()69 input[(outer * axisSize + i) * innerSize + inner]; in LogSoftmax()77 for (unsigned int i = 0u; i < axisSize; ++i) in LogSoftmax()79 const unsigned int index = (outer * axisSize + i) * innerSize + inner; in LogSoftmax()
24 const unsigned int axisSize = inputTensorInfo.GetShape()[uAxis]; in ArgMinMax() local31 in[outer * axisSize * innerElements + inner]; in ArgMinMax()34 for (unsigned int i = 1; i < axisSize; ++i) { in ArgMinMax()35 in[(outer * axisSize * innerElements) + (i * innerElements) + inner]; in ArgMinMax()
30 const unsigned int axisSize = inputShape[uAxis]; in Softmax() local37 unsigned int inputBeginIdx = outer * axisSize * innerSize; in Softmax()38 unsigned int inputEndIdx = inputBeginIdx + axisSize * innerSize; in Softmax()39 unsigned int outputBeginIdx = outer * axisSize * innerSize; in Softmax()
150 size_t axisSize = rmins.size(); in fakeQuantAttrsToType() local151 if (axisSize != rmaxs.size()) { in fakeQuantAttrsToType()153 << axisSize << " vs. " << rmaxs.size(), in fakeQuantAttrsToType()169 scales.reserve(axisSize); in fakeQuantAttrsToType()170 zeroPoints.reserve(axisSize); in fakeQuantAttrsToType()171 for (size_t axis = 0; axis != axisSize; ++axis) { in fakeQuantAttrsToType()
383 const int axisSize = armnn::numeric_cast<int>(inputShape[axis]); in GetStartForAxis() local386 start += (axisSize); in GetStartForAxis()389 return std::max(0, std::min(start, axisSize - 1)); in GetStartForAxis()417 const int axisSize = armnn::numeric_cast<int>(inputShape[axis]); in GetStopForAxis() local420 stop += axisSize; in GetStopForAxis()423 return m_Stride[axis] > 0 ? std::max(0, std::min(stop, axisSize)) : in GetStopForAxis()424 std::max(-1, std::min(stop, axisSize - 1)); in GetStopForAxis()
81 axisSize = header["axisSize"]84 axis.decompile(data[pos:pos+axisSize])86 pos += axisSize
28 SK_OT_USHORT axisSize; // Must be 0x0014 in v1.0 member
216 axisSize == 20 && /* Assumed in our code. */ in sanitize()424 HBUINT16 axisSize; /* The size in bytes of each VariationAxisRecord — member
2224 FT_UShort axisSize; member2309 FT_FRAME_USHORT ( axisSize ), in TT_Get_MM_Var()
768 axisSize