Home
last modified time | relevance | path

Searched refs:signature_inputs (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/python/saved_model/
Dsignature_def_utils_impl.py98 signature_inputs = {signature_constants.REGRESS_INPUTS: input_tensor_info}
106 signature_inputs, signature_outputs,
148 signature_inputs = {signature_constants.CLASSIFY_INPUTS: input_tensor_info}
165 signature_inputs, signature_outputs,
200 signature_inputs = {key: utils.build_tensor_info(tensor)
206 signature_inputs, signature_outputs,
252 signature_inputs = {key: utils.build_tensor_info(tensor)
263 signature_inputs, signature_outputs, method_name)
/external/tensorflow/tensorflow/lite/java/src/main/native/
Dnativeinterpreterwrapper_jni.cc174 (is_input ? interpreter->signature_inputs(method_name_ptr) in GetTensorIndexForSignature()
299 const jobjectArray signature_inputs = GetSignatureInputsOutputsList( in Java_org_tensorflow_lite_NativeInterpreterWrapper_getSignatureInputs() local
300 interpreter->signature_inputs(method_name_ptr), env); in Java_org_tensorflow_lite_NativeInterpreterWrapper_getSignatureInputs()
303 return signature_inputs; in Java_org_tensorflow_lite_NativeInterpreterWrapper_getSignatureInputs()
/external/tensorflow/tensorflow/lite/python/interpreter_wrapper/
Dinterpreter_wrapper.cc575 interpreter_->signature_inputs(sig_def_name->c_str()); in GetSignatureDefs()
604 const auto& inputs = interpreter_->signature_inputs(method_name); in SetInputTensorFromSignatureDefName()
/external/tensorflow/tensorflow/lite/
Dinterpreter.h307 const std::map<std::string, uint32_t>& signature_inputs( in signature_inputs() function
Dinterpreter_test.cc1859 EXPECT_THAT(interpreter_.signature_inputs(kMethodName), testing::Eq(inputs)); in TEST_F()