Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc223 Status RemoveBatchDimension(nvinfer1::Dims* dims) { in RemoveBatchDimension() function
3899 TF_RETURN_IF_ERROR(RemoveBatchDimension(&dims)); in ConvertBatchMatMul()