Searched defs:outer_size (Results 1 – 13 of 13) sorted by relevance
47 size_t outer_size = size + kGuardSize * 2; in Allocate() local102 const size_t outer_size = size_ + kGuardSize * 2; in ~PagedMemory() local
32 const int outer_size = in Softmax() local79 const int outer_size = in Softmax() local
475 const int outer_size = in L2Normalization() local500 const int outer_size = in L2Normalization() local1623 int64_t outer_size = 1; in Concatenation() local1676 int64_t outer_size = 1; in ConcatenationWithScaling() local1721 int outer_size = 1; in Pack() local1748 int outer_size = 1; in Unpack() local1787 int outer_size = 1; in PackWithScaling() local2086 const int outer_size = MatchingFlatSizeSkipDim( in LstmCell() local2247 int64_t outer_size = 1; in Split() local2546 const int outer_size = in LocalResponseNormalization() local[all …]
104 size_t outer_size = kMetaPageSize + size_ * 2 + kGuardSize; in ~SharedRingBuffer() local142 size_t outer_size = kMetaPageSize + size * 2 + kGuardSize; in Initialize() local
23 inline void L2Normalization(int32_t input_zero_point, int32_t outer_size, in L2Normalization()
25 int32_t diff_min, int32_t outer_size, int32_t depth, in LogSoftmax()
43 const int outer_size = in Softmax() local
113 const int outer_size = in Eval() local
867 const int outer_size = in LogSoftmaxEval() local
90 int64 outer_size = 1; in Compute() local
224 const int outer_size = in RunOneLogSoftmaxTest() local
2567 const int outer_size = in L2Normalization() local2595 const int outer_size = in L2Normalization() local3724 const int outer_size = MatchingFlatSizeSkipDim( in LstmCell() local4471 const int outer_size = in Softmax() local4669 const int outer_size = in LogSoftmax() local4722 const int outer_size = in LogSoftmax() local
2735 const int outer_size = in L2Normalization() local2812 const int outer_size = in L2Normalization() local3840 const int outer_size = MatchingFlatSizeSkipDim( in LstmCell() local4596 const int outer_size = in Softmax() local4807 const int outer_size = in LogSoftmax() local4983 const int outer_size = in LogSoftmax() local