Searched refs:GetWeightRange (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.h | 537 Status GetWeightRange(const TRT_ShapedWeights& weights, float* out_min,
|
| D | convert_nodes_test.cc | 552 Status GetWeightRange(const TRT_ShapedWeights& weights, float* out_min, in GetWeightRange() function in tensorflow::tensorrt::convert::ConverterTest
554 return converter_->GetWeightRange(weights, out_min, out_max); in GetWeightRange()
812 TF_EXPECT_OK(test->GetWeightRange(weights, &out_min, &out_max)); in TestGetWeightRange()
817 TEST_F(ConverterTest, GetWeightRange) { in TEST_F() argument
|
| D | convert_nodes.cc | 1193 Status Converter::GetWeightRange(const TRT_ShapedWeights& weights, in GetWeightRange() function in tensorflow::tensorrt::convert::Converter
1271 GetWeightRange(input.weights(), &min_range, &max_range)); in PrepareTensorForShape()
|