Searched refs:t_nchw_vect_c (Results 1 – 1 of 1) sorted by relevance
723 Tensor t_nchw_vect_c(DT_QINT8, TensorShape({2, 3, 5, 7, 4})); in TEST() local724 auto nchw_vect_c = t_nchw_vect_c.tensor<qint8, 5>(); in TEST()728 for (int n = 0; n < t_nchw_vect_c.shape().dim_size(0); ++n) { in TEST()729 for (int c = 0; c < t_nchw_vect_c.shape().dim_size(1); ++c) { in TEST()730 for (int h = 0; h < t_nchw_vect_c.shape().dim_size(2); ++h, ++val) { in TEST()732 for (int w = 0; w < t_nchw_vect_c.shape().dim_size(3); ++w) { in TEST()742 auto actual_nchw = t_nchw_vect_c.reinterpret_last_dimension<int32, 4>(); in TEST()743 const auto& const_t_nchw_vect_c = t_nchw_vect_c; in TEST()746 for (int n = 0; n < t_nchw_vect_c.shape().dim_size(0); ++n) { in TEST()747 for (int c = 0; c < t_nchw_vect_c.shape().dim_size(1); ++c) { in TEST()[all …]