Searched refs:node_mins (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/lite/quantization/ |
| D | quantization_config.cc | 48 std::vector<llvm::Optional<double>> node_mins; in ParseInputNodeQuantSpecs() local
56 node_mins.push_back(value); in ParseInputNodeQuantSpecs()
78 return GetInputNodeQuantSpecs(input_nodes, node_mins, node_maxs, final_type, in ParseInputNodeQuantSpecs()
84 const std::vector<llvm::Optional<double>>& node_mins, in GetInputNodeQuantSpecs() argument
90 if (node_mins.empty() || node_maxs.empty()) return false; in GetInputNodeQuantSpecs()
95 if (node_names.size() != node_mins.size() || in GetInputNodeQuantSpecs()
100 quant_specs->input_ranges.push_back({node_mins[i], node_maxs[i]}); in GetInputNodeQuantSpecs()
104 if (!node_mins.empty()) { in GetInputNodeQuantSpecs()
|
| D | quantization_config.h | 151 const std::vector<llvm::Optional<double>>& node_mins,
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/python/ |
| D | graphdef_to_tfl_flatbuffer.cc | 59 std::vector<llvm::Optional<double>> node_mins; in ConvertGraphDefToTFLiteFlatBuffer() local
65 &node_shapes, &node_mins, &node_maxs)); in ConvertGraphDefToTFLiteFlatBuffer()
|
| D | tf_tfl_flatbuffer_helpers.cc | 199 std::vector<llvm::Optional<double>>* node_mins, in PopulateQuantizationSpecs() argument
235 node_mins->push_back(min_max.first); in PopulateQuantizationSpecs()
238 node_mins->push_back(llvm::None); in PopulateQuantizationSpecs()
244 if (mlir::TFL::GetInputNodeQuantSpecs(*node_names, *node_mins, *node_maxs, in PopulateQuantizationSpecs()
|
| D | saved_model_to_tfl_flatbuffer.cc | 132 std::vector<llvm::Optional<double>> node_mins; in ConvertSavedModelToTFLiteFlatBuffer() local
138 &node_shapes, &node_mins, &node_maxs)); in ConvertSavedModelToTFLiteFlatBuffer()
|
| D | tf_tfl_flatbuffer_helpers.h | 45 std::vector<llvm::Optional<double>>* node_mins,
|