Searched defs:dim_stride (Results 1 – 8 of 8) sorted by relevance
176 int64_t dim_stride = inner_size; local283 int64_t dim_stride = inner_size; in host_softmax_backward() local
55 const size_t dim_stride = getTrailingDims(out, dim); in cat_out() local
80 const size_t dim_stride = in.strides()[dim]; in perform_topk() local
375 int64_t dim_stride = strides[i].template cast<IntegerAttr>().getInt(); in getPaddingValuesFromPadType() local453 int64_t dim_stride = strides[i].template cast<IntegerAttr>().getInt(); in getTransposeConv2dPaddingValues() local
52 int64_t dim_stride = inner_size; in log_softmax_kernel() local
262 const index_t dim_stride = inner_size; in cunn_SpatialSoftMaxForward() local320 const uint32_t dim_stride = inner_size; in cunn_SpatialSoftMaxBackward() local
131 int64_t dim_stride = strides.at(ndim - i); in unsqueeze_strides() local
699 int64_t dim_stride = inner_size; in _vec_softmax() local806 int64_t dim_stride = inner_size; in _vec_softmax() local