Searched defs:y_reshaped (Results 1 – 1 of 1) sorted by relevance
725 const Tensor y_reshaped = y.view({y.size(0), -1, y.size(-1)}); in jagged_dense_elementwise_dense_output_() local907 const Tensor y_reshaped = y.view({y.size(0), -1, y.size(-1)}); in jagged_dense_elementwise_jagged_output_() local1328 const Tensor y_reshaped = y.view({y.size(0), -1, y.size(-1)}); in jagged_dense_elementwise_jagged_output_opt_() local