Searched refs:runtime_inputs (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | model_builder_helper.cc | 219 int runtime_inputs, int outputs) { in CheckInputsOutputs() argument
222 if (runtime_inputs_from_model != runtime_inputs) { in CheckInputsOutputs()
224 "Expected ", runtime_inputs, " runtime input tensor(s), but node has ", in CheckInputsOutputs()
238 int runtime_inputs, int const_inputs, in CheckInputsConstsOutputs() argument
247 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 | 224 int runtime_inputs, in VerifyInputsConstsOutputs() argument
227 return CheckInputsConstsOutputs(context_, node, runtime_inputs, const_inputs, in VerifyInputsConstsOutputs()
|
| D | model_builder.cc | 350 const int runtime_inputs = in IsSupported() local
352 if (runtime_inputs > 2) { in IsSupported()
355 runtime_inputs, " runtime inputs.")); in IsSupported()
363 if (runtime_inputs == 1) { in IsSupported()
383 const int runtime_inputs = reader->GetNumberOfRuntimeInputs(); in Parse() local
384 if (runtime_inputs == 2) { in Parse()
410 const int runtime_inputs = in IsSupported() local
412 if (runtime_inputs > 2) { in IsSupported()
415 runtime_inputs, " runtime inputs.")); in IsSupported()
423 if (runtime_inputs == 1) { in IsSupported()
[all …]
|
| D | object_reader.h | 99 int runtime_inputs, int const_inputs,
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_execute.cc | 196 std::vector<xla::ExecutionInput>* runtime_inputs, in UpdateDynamicInputs() argument
198 TF_RET_CHECK(runtime_inputs->size() == compile_time_shapes.size()); in UpdateDynamicInputs()
206 auto& runtime_input = (*runtime_inputs)[i]; in UpdateDynamicInputs()
|