Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc2588 size_t output_shape_dims_count = weights.count(); in ConvertReshape() local
2595 output_nonbatch_dims.nbDims = output_shape_dims_count - 1; in ConvertReshape()
2596 for (int i = 1; i < output_shape_dims_count; i++) { in ConvertReshape()