Searched refs:num_outputs_ (Results 1 – 7 of 7) sorted by relevance
| /external/tflite-support/tensorflow_lite_support/cc/task/vision/ |
| D | image_classifier.cc | 162 num_outputs_ = TfLiteEngine::OutputCount(engine_->interpreter()); in CheckAndSetOutputs()
176 if (num_outputs_ != num_output_tensors) { in CheckAndSetOutputs()
182 num_outputs_, num_output_tensors), in CheckAndSetOutputs()
203 classification_heads_.reserve(num_outputs_); in CheckAndSetOutputs()
204 for (int output_index = 0; output_index < num_outputs_; ++output_index) { in CheckAndSetOutputs()
209 if (num_outputs_ != classification_heads_.size()) { in CheckAndSetOutputs()
214 num_outputs_, classification_heads_.size()), in CheckAndSetOutputs()
219 for (int i = 0; i < num_outputs_; ++i) { in CheckAndSetOutputs()
286 if (num_quantized_outputs > 0 && num_quantized_outputs != num_outputs_) { in CheckAndSetOutputs()
291 num_quantized_outputs, num_outputs_), in CheckAndSetOutputs()
[all …]
|
| D | image_classifier.h | 157 int num_outputs_; variable
|
| /external/tensorflow/tensorflow/js/ops/ |
| D | ts_op_gen.cc | 86 int num_outputs_; member in tensorflow::__anonc3e3999b0111::GenTypeScriptOp
90 : op_def_(op_def), api_def_(api_def), num_outputs_(0) {} in GenTypeScriptOp()
135 num_outputs_ = api_def_.out_arg_size(); in ProcessArgs()
189 if (num_outputs_ == 1) { in AddMethodSignature()
|
| /external/tensorflow/tensorflow/lite/delegates/coreml/builders/ |
| D | op_builder.h | 160 int num_outputs_ = 0; variable
|
| D | op_builder.cc | 218 auto tensor_id = TensorID(GetID(), num_outputs_++); in AddOutput()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime.h | 272 num_outputs_(num_outputs), in MultiDeviceFunctionData()
284 const int num_outputs_; member
|
| D | process_function_library_runtime.cc | 1048 output_devices->resize(data->num_outputs_); in GetOutputDevices()
1142 rets->resize(data->num_outputs_); in RunMultiDevice()
|