Searched refs:setAllowedFormats (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ | ||
| D | convert_nodes_test.cc | 383 void setAllowedFormats(nvinfer1::TensorFormats formats) override {} in setAllowedFormats() function in tensorflow::tensorrt::convert::FakeITensor |
| D | convert_nodes.cc | 817 void setAllowedFormats(nvinfer1::TensorFormats formats) override {} in setAllowedFormats() function in tensorflow::tensorrt::convert::TRT_TensorOrWeights::SimpleITensor |