Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dalgorithm_selector_test.cc47 EXPECT_FALSE(sel72.AllowShuffleAlgorithm(0, nvinfer1::DataType::kFLOAT, in TEST()
51 EXPECT_TRUE(sel72.AllowShuffleAlgorithm(0, nvinfer1::DataType::kHALF, in TEST()
53 EXPECT_TRUE(sel72.AllowShuffleAlgorithm(0, nvinfer1::DataType::kINT32, in TEST()
55 EXPECT_TRUE(sel72.AllowShuffleAlgorithm(0, nvinfer1::DataType::kFLOAT, in TEST()
71 EXPECT_FALSE(sel80.AllowShuffleAlgorithm(0, nvinfer1::DataType::kINT8, in TEST()
75 EXPECT_TRUE(sel80.AllowShuffleAlgorithm(0, nvinfer1::DataType::kHALF, in TEST()
77 EXPECT_TRUE(sel80.AllowShuffleAlgorithm(0, nvinfer1::DataType::kINT32, in TEST()
79 EXPECT_TRUE(sel80.AllowShuffleAlgorithm(0, nvinfer1::DataType::kFLOAT, in TEST()
81 EXPECT_TRUE(sel80.AllowShuffleAlgorithm(0, nvinfer1::DataType::kINT8, in TEST()
83 EXPECT_TRUE(sel80.AllowShuffleAlgorithm(0, nvinfer1::DataType::kINT8, in TEST()
H A Dalgorithm_selector.cc117 bool AlgorithmSelectorImpl::AllowShuffleAlgorithm( in AllowShuffleAlgorithm() function in tensorflow::tensorrt::convert::AlgorithmSelectorImpl
204 return selector_.AllowShuffleAlgorithm( in AlgorithmPolicy()
H A Dalgorithm_selector.h53 bool AllowShuffleAlgorithm(TacticID tactic, nvinfer1::DataType input_dtype,