Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc2405 const nvinfer1::Dims actual_output_dims = output.tensor()->getDimensions(); in TEST_F() local
2406 ExpectTrtDimsEqualsArray(expected_output_dims, actual_output_dims); in TEST_F()
2408 std::vector<float> input_vec(TrtTensorDimsNumElements(actual_output_dims) * in TEST_F()