Home
last modified time | relevance | path

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

12

/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.py68 x_tensor_info = utils.build_tensor_info(x)
77 x_tensor_info = utils.build_tensor_info(x)
92 x_tensor_info = utils.build_tensor_info(x)
113 utils.build_tensor_info(x)
118 tensor_info = utils.build_tensor_info(expected)
126 tensor_info = utils.build_tensor_info(expected)
136 tensor_info = utils.build_tensor_info(expected)
145 tensor_info = utils.build_tensor_info(expected)
159 tensor_info = utils.build_tensor_info(unscoped)
175 tensor_info = utils.build_tensor_info(expected)
Dloader_test.py63 "foo_input": utils.build_tensor_info(x)
64 }, {"foo_output": utils.build_tensor_info(z)})
66 "bar_x": utils.build_tensor_info(x),
67 "bar_y": utils.build_tensor_info(y)
68 }, {"bar_z": utils.build_tensor_info(z)})
259 {"foo_input": utils.build_tensor_info(x)},
260 {"foo_output": utils.build_tensor_info(z)})
Dload_v1_in_v2_test.py145 {"start": utils_impl.build_tensor_info(start)},
146 {"output": utils_impl.build_tensor_info(output)})
173 {"first_start": utils_impl.build_tensor_info(start)},
174 {"first_output": utils_impl.build_tensor_info(
180 {"second_start": utils_impl.build_tensor_info(start)},
181 {"second_output": utils_impl.build_tensor_info(
394 {}, dict(value=utils_impl.build_tensor_info(output)))})
423 {}, dict(value=utils_impl.build_tensor_info(output)))})
683 "input1": utils_impl.build_tensor_info(input1),
684 "input2": utils_impl.build_tensor_info(input2)
[all …]
Dutils.py24 from tensorflow.python.saved_model.utils_impl import build_tensor_info
Dsignature_def_utils_test.py50 input_name: utils.build_tensor_info(tensor)
54 output_name: utils.build_tensor_info(tensor)
69 x_tensor_info = utils.build_tensor_info(x)
74 y_tensor_info = utils.build_tensor_info(y)
Dutils_impl.py51 def build_tensor_info(tensor): function
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model/
Dmulti_arguments_results_v1.py61 tensor_info_x = tf.compat.v1.saved_model.utils.build_tensor_info(x)
62 tensor_info_y = tf.compat.v1.saved_model.utils.build_tensor_info(y)
63 tensor_info_s = tf.compat.v1.saved_model.utils.build_tensor_info(s)
64 tensor_info_t = tf.compat.v1.saved_model.utils.build_tensor_info(t)
Ddefun_export.py46 tensor_info_x = tf.compat.v1.saved_model.utils.build_tensor_info(x)
47 tensor_info_y = tf.compat.v1.saved_model.utils.build_tensor_info(y)
48 tensor_info_z = tf.compat.v1.saved_model.utils.build_tensor_info(z)
Dbasic_v1.py59 tensor_info_x = tf.compat.v1.saved_model.utils.build_tensor_info(x)
60 tensor_info_r = tf.compat.v1.saved_model.utils.build_tensor_info(r)
Dimport_restore_v1.py67 tensor_info_x = tf.compat.v1.saved_model.utils.build_tensor_info(x)
68 tensor_info_r = tf.compat.v1.saved_model.utils.build_tensor_info(r)
Dshared_variable_v1.py51 tensor_info_x = tf.saved_model.utils.build_tensor_info(x)
52 tensor_info_r = tf.saved_model.utils.build_tensor_info(r)
Dduplicate_method_names_v1.py45 tensor_info_s = tf.compat.v1.saved_model.utils.build_tensor_info(s)
46 tensor_info_t = tf.compat.v1.saved_model.utils.build_tensor_info(t)
Dremove_init_variable_v1.py60 tensor_info_x = tf.compat.v1.saved_model.utils.build_tensor_info(x)
61 tensor_info_r = tf.compat.v1.saved_model.utils.build_tensor_info(r)
Dhash_table_v1.py79 tensor_info_x = tf.compat.v1.saved_model.utils.build_tensor_info(x)
80 tensor_info_r = tf.compat.v1.saved_model.utils.build_tensor_info(r)
Dhash_table_asset_v1.py67 tensor_info_x = tf.compat.v1.saved_model.utils.build_tensor_info(x)
68 tensor_info_r = tf.compat.v1.saved_model.utils.build_tensor_info(r)
Dmulti_variables_v1.py52 tensor_info_z = tf.compat.v1.saved_model.utils.build_tensor_info(z)
Dcontrol_flow_duplicate_v1.py52 tensor_info_result = tf.compat.v1.saved_model.utils.build_tensor_info(result)
Dno_input_shape_v1.py54 tensor_info_r = tf.compat.v1.saved_model.utils.build_tensor_info(r)
Dcontrol_flow_upgrade_legacy_v1.py44 tensor_info_result = tf.compat.v1.saved_model.utils.build_tensor_info(result)
/external/tensorflow/tensorflow/python/compiler/tensorrt/test/testdata/
Dgen_tftrt_model.py112 "input1": utils.build_tensor_info(input1),
113 "input2": utils.build_tensor_info(input2)
115 outputs={"output": utils.build_tensor_info(out)},
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.saved_model.utils.pbtxt4 name: "build_tensor_info"
Dtensorflow.saved_model.pbtxt160 name: "build_tensor_info"
/external/tensorflow/tensorflow/examples/speech_commands/
Dfreeze.py188 'input': tf.compat.v1.saved_model.utils.build_tensor_info(input_tensor)
191 'output': tf.compat.v1.saved_model.utils.build_tensor_info(output_tensor)
/external/tensorflow/tensorflow/lite/python/
Dconvert_saved_model_test.py221 sig_input_tensor = saved_model.utils.build_tensor_info(in_tensor)
223 sig_output_tensor = saved_model.utils.build_tensor_info(out_tensor)

12