Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc2593 const nvinfer1::Dims input_nonbatch_dims = input_tensor.GetTrtDims(); in ConvertReshape() local
2601 << ", input_nonbatch_dims=" << DebugString(input_nonbatch_dims) in ConvertReshape()
2612 !AreDimsStaticWithSameSize(input_nonbatch_dims, output_nonbatch_dims, in ConvertReshape()
2619 ", ", DebugString(input_nonbatch_dims), in ConvertReshape()