Searched defs:expand_size (Results 1 – 5 of 5) sorted by relevance
9 inline c10::MaybeOwned<TensorBase> expand_size( in expand_size() function
393 inline c10::MaybeOwned<Tensor> expand_size( in expand_size() function407 inline c10::MaybeOwned<Tensor> expand_size( in expand_size() function
582 auto expand_size = input_values.sizes().vec(); in _sparse_sum_backward_cuda() local602 auto expand_size = input_values.sizes().vec(); in _sparse_sum_backward_cuda() local
398 def _avg_pool(name, expand_size): argument
1801 auto expand_size = input_values.sizes().vec(); in _sparse_sum_backward_cpu() local1823 auto expand_size = input_values.sizes().vec(); in _sparse_sum_backward_cpu() local