Searched refs:TfDataTypeToTrt (Results 1 – 1 of 1) sorted by relevance
86 nvinfer1::DataType TfDataTypeToTrt(DataType tf_dtype) { in TfDataTypeToTrt() function981 EXPECT_EQ(TfDataTypeToTrt(dtype), tensor->getType()) in TEST_F()982 << "Expected " << DebugString(TfDataTypeToTrt(dtype)) << " vs. actual " in TEST_F()1122 {data.name, data.name, TfDataTypeToTrt(data.tensor.dtype())}); in BuildAndRun()1671 TfDataTypeToTrt(dtype)); in TestConvertBiasAdd()1777 TfDataTypeToTrt(dtype)); in TestBinaryTensorOpWeightNoBroadcast()1833 TfDataTypeToTrt(dtype)); in TestBinaryTensorOpWeightWithChannelWiseBroadcast()1868 TfDataTypeToTrt(dtype)); in TestBinaryTensorOpWeightWithUniformlyBroadcast()1904 TfDataTypeToTrt(dtype)); in TestBinaryTensorOpWeightFallback()1957 TfDataTypeToTrt(dtype)); in TestBinaryTensorOpTensor()[all …]