Searched defs:embedding_features (Results 1 – 2 of 2) sorted by relevance
477 int64_t embedding_features, in _embedding_bag_per_sample_weights_backward_kernel()520 auto embedding_features = grad.size(1); in _embedding_bag_per_sample_weights_backward_cuda() local
1657 auto embedding_features = grad.sizes()[1]; in _embedding_bag_per_sample_weights_backward_cpu_template() local