Searched refs:shuffle_dims (Results 1 – 1 of 1) sorted by relevance
209 DSizes<Eigen::DenseIndex, 4> shuffle_dims(0, 1, 2, 3); in test_nested_ops_with_ref() local216 Tensor<float, 4> padded = ref.shuffle(shuffle_dims).pad(trivial); in test_nested_ops_with_ref()