Searched refs:runtime_inputs (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | model_builder_helper.cc | 221 int runtime_inputs, int outputs) { in CheckInputsOutputs() argument
224 if (runtime_inputs_from_model != runtime_inputs) { in CheckInputsOutputs()
226 "Expected ", runtime_inputs, " runtime input tensor(s), but node has ", in CheckInputsOutputs()
240 int runtime_inputs, int const_inputs, in CheckInputsConstsOutputs() argument
249 return CheckInputsOutputs(context, tflite_node, runtime_inputs, outputs); in CheckInputsConstsOutputs()
|
| D | model_builder_helper.h | 68 int runtime_inputs, int outputs);
72 int runtime_inputs, int const_inputs,
|
| D | object_reader.cc | 237 int runtime_inputs, in VerifyInputsConstsOutputs() argument
240 return CheckInputsConstsOutputs(context_, node, runtime_inputs, const_inputs, in VerifyInputsConstsOutputs()
|
| D | object_reader.h | 136 int runtime_inputs, int const_inputs,
|
| D | model_builder.cc | 589 const int runtime_inputs = reader->GetNumberOfRuntimeInputs(); in Parse() local
590 if (runtime_inputs == 2) { in Parse()
640 const int runtime_inputs = reader->GetNumberOfRuntimeInputs(); in ReadAttributes() local
641 if (runtime_inputs == 1) { in ReadAttributes()
842 const int runtime_inputs = reader->GetNumberOfRuntimeInputs(); in Parse() local
843 if (runtime_inputs == 2) { in Parse()
940 const int runtime_inputs = reader->GetNumberOfRuntimeInputs(); in Parse() local
941 if (runtime_inputs == 0) { in Parse()
959 if (runtime_inputs == 1) { in Parse()
2573 const int runtime_inputs = reader->GetNumberOfRuntimeInputs(); in Parse() local
[all …]
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_execute.cc | 198 std::vector<xla::ExecutionInput>* runtime_inputs, in UpdateDynamicInputs() argument
200 TF_RET_CHECK(runtime_inputs->size() == compile_time_shapes.size()); in UpdateDynamicInputs()
208 auto& runtime_input = (*runtime_inputs)[i]; in UpdateDynamicInputs()
|
| /external/tensorflow/tensorflow/lite/tools/versioning/ |
| D | gpu_compatibility.cc | 162 const int runtime_inputs = GetNumberOfRuntimeInputs(op_sig); in CheckConvoultionInputOutput() local
163 if (runtime_inputs > 2) { in CheckConvoultionInputOutput()
166 runtime_inputs, " runtime inputs.")); in CheckConvoultionInputOutput()
174 if (runtime_inputs == 1) { in CheckConvoultionInputOutput()
|