Home
last modified time | relevance | path

Searched refs:input_model (Results 1 – 25 of 31) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/strip_buffers/
H A Dstripping_lib.cc110 const Model* input_model, in StripWeightsFromFlatbuffer() argument
113 if (input_model->subgraphs()->size() != 1) { in StripWeightsFromFlatbuffer()
115 << input_model->subgraphs()->size(); in StripWeightsFromFlatbuffer()
124 const SubGraph* input_subgraph = (*input_model->subgraphs())[0]; in StripWeightsFromFlatbuffer()
133 const int num_buffers = input_model->buffers()->size(); in StripWeightsFromFlatbuffer()
150 if (auto* buffer = (*input_model->buffers())[tensor->buffer]) { in StripWeightsFromFlatbuffer()
164 for (const Buffer* buffer : *(input_model->buffers())) { in StripWeightsFromFlatbuffer()
181 for (const OperatorCode* opcode : *(input_model->operator_codes())) { in StripWeightsFromFlatbuffer()
201 auto input_model = FlatBufferModel::BuildFromBuffer(input_flatbuffer.data(), in StripWeightsFromFlatbuffer() local
205 if (StripWeightsFromFlatbuffer(input_model->GetModel(), &builder) != in StripWeightsFromFlatbuffer()
[all …]
H A Dreconstitute_buffers_into_fb.cc52 auto input_model = in Main() local
55 if (!FlatbufferHasStrippedWeights(input_model->GetModel())) { in Main()
61 if (ReconstituteConstantTensorsIntoFlatbuffer(input_model->GetModel(), in Main()
67 << input_model->allocation()->bytes() / 1000.0; in Main()
H A Dstripping_lib.h33 const Model* input_model,
48 const Model* input_model,
59 bool FlatbufferHasStrippedWeights(const Model* input_model);
H A Dstrip_buffers_from_fb.cc52 auto input_model = in Main() local
57 if (StripWeightsFromFlatbuffer(input_model->GetModel(), &builder) != in Main()
63 << input_model->allocation()->bytes() / 1000.0; in Main()
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/tflite/
H A Dimport.cc34 void LoadTensorsTable(const ::tflite::Model& input_model, in LoadTensorsTable() argument
37 auto tensors = (*input_model.subgraphs())[0]->tensors(); in LoadTensorsTable()
44 void LoadOperatorsTable(const ::tflite::Model& input_model, in LoadOperatorsTable() argument
46 auto opcodes = input_model.operator_codes(); in LoadOperatorsTable()
59 void ImportTensors(const ::tflite::Model& input_model, Model* model) { in ImportTensors() argument
60 auto tensors = (*input_model.subgraphs())[0]->tensors(); in ImportTensors()
61 auto* buffers = input_model.buffers(); in ImportTensors()
105 const ::tflite::Model& input_model, in ImportOperators()
110 auto ops = (*input_model.subgraphs())[0]->operators(); in ImportOperators()
173 const ::tflite::Model& input_model, in ImportIOTensors()
[all …]
H A Dimport.h41 void LoadTensorsTable(const ::tflite::Model &input_model,
43 void LoadOperatorsTable(const ::tflite::Model &input_model,
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/
H A Dquantize_weights.cc396 flatbuffers::FlatBufferBuilder* builder, const Model* input_model, in QuantizeWeightsInt8() argument
401 model.reset(input_model->UnPack());
524 const Model* input_model) { in QuantizeWeightsFloat16() argument
526 model.reset(input_model->UnPack()); in QuantizeWeightsFloat16()
602 const Model* input_model, in QuantizeWeights() argument
610 builder, input_model, weights_min_num_elements, use_hybrid_evaluation); in QuantizeWeights()
613 return QuantizeWeightsInt8(builder, input_model, use_hybrid_evaluation, in QuantizeWeights()
620 const Model* input_model, in QuantizeWeights() argument
624 return mlir::lite::QuantizeWeights(builder, input_model, in QuantizeWeights()
628 return QuantizeWeightsInt8(builder, input_model, true, in QuantizeWeights()
[all …]
H A Dquantize_weights_portable.cc382 flatbuffers::FlatBufferBuilder* builder, const Model* input_model, in QuantizeWeightsInt8() argument
387 model.reset(input_model->UnPack());
510 const Model* input_model) { in QuantizeWeightsFloat16() argument
512 model.reset(input_model->UnPack()); in QuantizeWeightsFloat16()
588 const Model* input_model, in QuantizeWeights() argument
599 return QuantizeWeightsInt8(builder, input_model, use_hybrid_evaluation, in QuantizeWeights()
606 const Model* input_model, in QuantizeWeights() argument
614 return QuantizeWeightsInt8(builder, input_model, true, in QuantizeWeights()
620 const Model* input_model, BufferType quant_type, in QuantizeWeights() argument
632 return QuantizeWeightsInt8(builder, input_model, true, in QuantizeWeights()
[all …]
H A Dquantize_weights.h63 flatbuffers::FlatBufferBuilder* builder, const Model* input_model,
71 flatbuffers::FlatBufferBuilder* builder, const Model* input_model,
77 flatbuffers::FlatBufferBuilder* builder, const Model* input_model,
85 flatbuffers::FlatBufferBuilder* builder, const Model* input_model,
98 flatbuffers::FlatBufferBuilder* builder, const Model* input_model,
H A Dquantize_model.h38 ModelT* input_model, ErrorReporter* error_reporter);
45 ModelT* input_model, const TensorType& input_type,
54 ModelT* input_model, const TensorType& input_type,
63 ModelT* input_model, const TensorType& input_type,
H A Dquantization_wrapper_utils.cc65 auto input_model = FlatBufferModel::BuildFromFile(path.c_str()); in LoadModel() local
66 if (!input_model) { in LoadModel()
69 auto readonly_model = input_model->GetModel(); in LoadModel()
H A Dmodify_model_interface.h55 flatbuffers::FlatBufferBuilder* builder, const Model* input_model,
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/lite/
H A Dquantize_weights.cc68 const tflite::Model* input_model) { in CreateMutableModelFromFile() argument
70 input_model->UnPackTo(copied_model.get(), nullptr); in CreateMutableModelFromFile()
78 flatbuffers::FlatBufferBuilder* builder, const tflite::Model* input_model, in QuantizeWeights() argument
98 input_builder, CreateMutableModelFromFile(input_model).get()); in QuantizeWeights()
171 const tflite::Model* input_model, in QuantizeWeights() argument
176 builder, input_model, &error_reporter, in QuantizeWeights()
188 const tflite::Model* input_model, in QuantizeWeights() argument
200 return QuantizeWeights(builder, input_model, &error_reporter, inference_type, in QuantizeWeights()
210 const tflite::Model* input_model, in QuantizeWeights() argument
222 builder, input_model, &error_reporter, inference_type, in QuantizeWeights()
H A Dquantize_weights.h61 const tflite::Model* input_model,
73 const tflite::Model* input_model,
78 const tflite::Model* input_model,
83 const tflite::Model* input_model,
H A Dquantize_model.cc51 const tflite::ModelT& input_model, const tflite::TensorType& input_type, in QuantizeModel() argument
75 tflite::Model::Pack(input_builder, &input_model); in QuantizeModel()
H A Dquantize_model.h42 const tflite::ModelT& input_model, const tflite::TensorType& input_type,
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/
H A Dgen_op_registration_main.cc138 void AddOpsFromModel(const std::string& input_model, in AddOpsFromModel() argument
141 std::ifstream fin(input_model); in AddOpsFromModel()
165 for (const std::string& input_model : models) { in main() local
166 AddOpsFromModel(input_model, &builtin_ops, &custom_ops); in main()
/aosp_15_r20/external/libopus/dnn/training_tf2/
H A Dtrain_rdovae.py84 input_model = args.quantize variable
96 input_model = args.retrain variable
139 model.load_weights(input_model)
H A Dtrain_plc.py86 input_model = args.quantize variable
98 input_model = args.retrain variable
185 model.load_weights(input_model)
H A Dtrain_lpcnet.py116 input_model = args.quantize variable
128 input_model = args.retrain variable
192 model.load_weights(input_model)
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/sparsity/
H A Dsparsify_model.cc39 TfLiteStatus SparsifyModel(const tflite::ModelT& input_model, in SparsifyModel() argument
49 tflite::Model::Pack(input_builder, &input_model); in SparsifyModel()
H A Dsparsify_model.h29 TfLiteStatus SparsifyModel(const tflite::ModelT& input_model,
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/g3doc/
H A Dquantize_weights.md64 ::tflite::Model* input_model = ...;
66 TfLiteStatus status = ::tflite::optimize::QuantizeWeights(&builder, input_model);
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/
H A Dtf_to_tfl_flatbuffer.cc160 const ::tflite::Model* input_model = ::tflite::GetModel(buffer); in ApplyDynamicRangeQuantizationFromOldQuantizer() local
177 &q_builder, input_model, quantized_type, use_updated_hybrid_scheme, in ApplyDynamicRangeQuantizationFromOldQuantizer()
/aosp_15_r20/external/pytorch/torch/csrc/jit/mobile/compatibility/
H A Dbackport_manager.cpp171 std::stringstream& input_model, in update_bytecode_version() argument
173 PyTorchStreamReader reader_bytecode(&input_model); in update_bytecode_version()

12