Searched refs:fused_tensors (Results 1 – 2 of 2) sorted by relevance
237 std::vector<Tensor> fused_tensors; in CompareFusedAndSeparate() local238 TF_ASSERT_OK(session->Run({}, {"fused_conv"}, {}, &fused_tensors)); in CompareFusedAndSeparate()240 test::ExpectClose(unfused_tensors[0], fused_tensors[0]); in CompareFusedAndSeparate()290 std::vector<Tensor> fused_tensors; in CompareFusedPadOnlyAndSeparate() local291 TF_ASSERT_OK(session->Run({}, {"fused_conv"}, {}, &fused_tensors)); in CompareFusedPadOnlyAndSeparate()293 test::ExpectClose(unfused_tensors[0], fused_tensors[0]); in CompareFusedPadOnlyAndSeparate()
1877 std::set<nvinfer1::ITensor*> fused_tensors; in MaybeApplyQuantizationRanges() local1935 fused_tensors.insert(fused_layer->getOutput(i)); in MaybeApplyQuantizationRanges()1946 fused_tensors.find(tensor) == fused_tensors.end()) { in MaybeApplyQuantizationRanges()