Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc2128 const int reshape_batch_dim = weights_ptr[0]; in ConvertReshape() local
2141 if (reshape_batch_dim == -1) { // Other dims of the shape must be fixed. in ConvertReshape()
2146 } else if (reshape_batch_dim != input_batch_dim) { in ConvertReshape()
2159 << "\nreshape_batch_dim=" << reshape_batch_dim in ConvertReshape()