Searched refs:t_nchw_vect_c (Results 1 – 1 of 1) sorted by relevance
756 Tensor t_nchw_vect_c(DT_QINT8, TensorShape({2, 3, 5, 7, 4})); in TEST() local757 auto nchw_vect_c = t_nchw_vect_c.tensor<qint8, 5>(); in TEST()761 for (int n = 0; n < t_nchw_vect_c.shape().dim_size(0); ++n) { in TEST()762 for (int c = 0; c < t_nchw_vect_c.shape().dim_size(1); ++c) { in TEST()763 for (int h = 0; h < t_nchw_vect_c.shape().dim_size(2); ++h, ++val) { in TEST()765 for (int w = 0; w < t_nchw_vect_c.shape().dim_size(3); ++w) { in TEST()775 auto actual_nchw = t_nchw_vect_c.reinterpret_last_dimension<int32, 4>(); in TEST()776 const auto& const_t_nchw_vect_c = t_nchw_vect_c; in TEST()779 for (int n = 0; n < t_nchw_vect_c.shape().dim_size(0); ++n) { in TEST()780 for (int c = 0; c < t_nchw_vect_c.shape().dim_size(1); ++c) { in TEST()[all …]