Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc394 const int max_nb_dims = nvinfer1::Dims::MAX_DIMS + 1; in GetTrtBroadcastShape() local
400 std::fill(output_dims_array, output_dims_array + max_nb_dims, 1); in GetTrtBroadcastShape()
436 int output_l[max_nb_dims], output_r[max_nb_dims]; in GetTrtBroadcastShape()