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()
|