Searched defs:weight_stride1 (Results 1 – 2 of 2) sorted by relevance
65 int64_t featureSize, int64_t weight_stride0, int64_t weight_stride1, in EmbeddingBag_updateOutputKernel_max()115 int64_t featureSize, int64_t weight_stride0, int64_t weight_stride1, in EmbeddingBag_updateOutputKernel_sum_mean()473 const scalar_t* weight, int64_t weight_stride0, int64_t weight_stride1, in _embedding_bag_per_sample_weights_backward_kernel()
1091 auto weight_stride1 = weight.strides()[1]; in embedding_bag_cpu_max_out() local1696 auto weight_stride1 = weight.strides()[1]; in _embedding_bag_per_sample_weights_backward_cpu_template() local