Searched defs:num_jagged_dim (Results 1 – 1 of 1) sorted by relevance
583 const int num_jagged_dim = dense_tensor.dim() - 2; in check_shape_and_partition_() local707 const int num_jagged_dim = y.dim() - 2; in jagged_dense_elementwise_dense_output_() local894 const int num_jagged_dim = y.dim() - 2; in jagged_dense_elementwise_jagged_output_() local1173 const int& num_jagged_dim, in jagged_dense_dense_elementwise_jagged_output_matches_opt()1315 const int num_jagged_dim = y.dim() - 2; in jagged_dense_elementwise_jagged_output_opt_() local1419 const size_t num_jagged_dim = offsets.size(); in _fbgemm_jagged_to_padded_dense_forward() local