Home
last modified time | relevance | path

Searched refs:tensor_dim (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc1833 const auto tensor_dim = tensor->getDimensions(); in ConvertConv2DHelper() local
1838 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() local
2687 {static_cast<int>(tensor_dim.d[1]), static_cast<int>(tensor_dim.d[2])}); in ConvertPool()