Searched refs:tensor_dim (Results 1 – 1 of 1) sorted by relevance
1833 const auto tensor_dim = tensor->getDimensions(); in ConvertConv2DHelper() local1838 const int num_groups = (group == 0) ? tensor_dim.d[0] : group; in ConvertConv2DHelper()1875 input_dims = {static_cast<int>(tensor_dim.d[1]), in ConvertConv2DHelper()1876 static_cast<int>(tensor_dim.d[2])}; in ConvertConv2DHelper()2679 auto tensor_dim = tensor->getDimensions(); in ConvertPool() local2687 {static_cast<int>(tensor_dim.d[1]), static_cast<int>(tensor_dim.d[2])}); in ConvertPool()