Searched refs:GetWeightRange (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.h | 598 Status GetWeightRange(const TRT_ShapedWeights& weights, float* out_min,
|
| D | convert_nodes_test.cc | 770 Status GetWeightRange(const TRT_ShapedWeights& weights, float* out_min, in GetWeightRange() function in tensorflow::tensorrt::convert::ConverterTest
772 return converter_->GetWeightRange(weights, out_min, out_max); in GetWeightRange()
1062 TF_EXPECT_OK(test->GetWeightRange(weights, &out_min, &out_max)); in TestGetWeightRange()
1067 TEST_F(ConverterTest, GetWeightRange) { in TEST_F() argument
|
| D | convert_nodes.cc | 1629 Status Converter::GetWeightRange(const TRT_ShapedWeights& weights, in GetWeightRange() function in tensorflow::tensorrt::convert::Converter
1757 converter->GetWeightRange(input.weights(), &min_range, &max_range)); in PrepareTensorForShape()
|