Searched defs:inner_dense_size (Results 1 – 1 of 1) sorted by relevance
566 const int inner_dense_size = dense_tensor.size(-1); in check_shape_and_partition_() local651 const int inner_dense_size = y.size(2); in __launch_bounds__() local803 const int inner_dense_size = y_0.size(2); in __launch_bounds__() local