MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / tensor

Method tensor

tensorflow/compiler/tf2tensorrt/convert/convert_nodes.cc:749–752  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

747}
748
749ITensorProxyPtr TRT_TensorOrWeights::tensor() const {
750 CHECK(is_tensor());
751 return tensor_proxy_ptr_;
752}
753
754nvinfer1::Dims TRT_TensorOrWeights::GetTrtDims() const {
755 if (is_tensor()) {

Callers 15

ToLiteralMethod · 0.45
TEST_FFunction · 0.45
mainFunction · 0.45
matchAndRewriteMethod · 0.45
VerifyFunction · 0.45
ConvertAttributeValueMethod · 0.45
matchAndRewriteMethod · 0.45
ComputeMethod · 0.45
PopulateOutputsMethod · 0.45

Calls 1

is_tensorFunction · 0.50

Tested by 15

TEST_FFunction · 0.36
TESTFunction · 0.36
TEST_FFunction · 0.36
TestMatMulHelperFunction · 0.36
TestConvertBiasAddFunction · 0.36
TestBinaryOpFunction · 0.36
TestAddNFunction · 0.36
TestConvertSquareFunction · 0.36
TestConvertGatherFunction · 0.36
TestConvertConcatFunction · 0.36
TestConvertSplitFunction · 0.36
TestConvertUnpackFunction · 0.36