Home
last modified time | relevance | path

Searched refs:build_tensor_info (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/python/saved_model/
Dsignature_def_utils_impl.py95 input_tensor_info = utils.build_tensor_info(examples)
100 output_tensor_info = utils.build_tensor_info(predictions)
145 input_tensor_info = utils.build_tensor_info(examples)
152 classes_tensor_info = utils.build_tensor_info(classes)
158 scores_tensor_info = utils.build_tensor_info(scores)
200 signature_inputs = {key: utils.build_tensor_info(tensor)
202 signature_outputs = {key: utils.build_tensor_info(tensor)
252 signature_inputs = {key: utils.build_tensor_info(tensor)
258 sig_out = {key: utils.build_tensor_info(tensor)
Dutils_test.py64 x_tensor_info = utils.build_tensor_info(x)
73 x_tensor_info = utils.build_tensor_info(x)
89 utils.build_tensor_info(x)
94 tensor_info = utils.build_tensor_info(expected)
102 tensor_info = utils.build_tensor_info(expected)
112 tensor_info = utils.build_tensor_info(expected)
126 tensor_info = utils.build_tensor_info(unscoped)
142 tensor_info = utils.build_tensor_info(expected)
Dloader_test.py56 "foo_input": utils.build_tensor_info(x)
57 }, {"foo_output": utils.build_tensor_info(z)})
59 "bar_x": utils.build_tensor_info(x),
60 "bar_y": utils.build_tensor_info(y)
61 }, {"bar_z": utils.build_tensor_info(z)})
238 {"foo_input": utils.build_tensor_info(x)},
239 {"foo_output": utils.build_tensor_info(z)})
Dload_v1_in_v2_test.py121 {"first_start": utils_impl.build_tensor_info(start)},
122 {"first_output": utils_impl.build_tensor_info(
128 {"second_start": utils_impl.build_tensor_info(start)},
129 {"second_output": utils_impl.build_tensor_info(
Dsignature_def_utils_test.py50 input_name: utils.build_tensor_info(tensor)
54 output_name: utils.build_tensor_info(tensor)
66 x_tensor_info = utils.build_tensor_info(x)
71 y_tensor_info = utils.build_tensor_info(y)
Dutils.py24 from tensorflow.python.saved_model.utils_impl import build_tensor_info
Dutils_impl.py47 def build_tensor_info(tensor): function
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.saved_model.utils.pbtxt4 name: "build_tensor_info"
Dtensorflow.saved_model.pbtxt144 name: "build_tensor_info"
/external/tensorflow/tensorflow/lite/python/
Dconvert_saved_model_test.py298 sig_input_tensor = saved_model.utils.build_tensor_info(in_tensor)
300 sig_output_tensor = saved_model.utils.build_tensor_info(out_tensor)
/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dtrt_convert_test.py158 inputs={"myinput": utils.build_tensor_info(inp)},
159 outputs={"myoutput": utils.build_tensor_info(out)},