Home
last modified time | relevance | path

Searched refs:input_names (Results 1 – 25 of 70) sorted by relevance

123

/external/tensorflow/tensorflow/tools/graph_transforms/
Dfold_constants_lib.cc128 std::unordered_set<TensorId, TensorId::Hasher> input_names; in RemoveUnusedNodes() local
129 for (const string& input : context.input_names) { in RemoveUnusedNodes()
130 input_names.insert(ParseTensorName(input)); in RemoveUnusedNodes()
151 if (input_names.count(id) > 0) { in RemoveUnusedNodes()
161 for (const TensorId& id : input_names) { in RemoveUnusedNodes()
217 std::vector<TensorId> input_names; in FoldConstants() local
218 input_names.reserve(context.input_names.size()); in FoldConstants()
219 std::transform(context.input_names.begin(), context.input_names.end(), in FoldConstants()
220 std::back_inserter(input_names), in FoldConstants()
227 std::sort(input_names.begin(), input_names.end(), compare); in FoldConstants()
[all …]
Dremove_nodes_test.cc94 context.input_names = {}; in TestRemoveNodes()
149 context.input_names = {}; in TestRemoveOutputNodes()
198 context.input_names = {}; in TestRemoveChainedNodes()
248 context.input_names = {}; in TestRemoveMultipleInputs()
Dbackports_test.cc84 context.input_names = {}; in TestBackportConcatV2()
156 context.input_names = {}; in TEST()
187 context.input_names = {}; in TEST()
Dinsert_logging_test.cc70 context.input_names = {}; in TestInsertLogging()
117 context.input_names = {}; in TestInsertLoggingByOpType()
166 context.input_names = {}; in TestInsertLoggingByPrefix()
Dset_device_test.cc86 context.input_names = {}; in TEST()
107 context.input_names = {}; in TEST()
Dremove_attribute_test.cc84 context.input_names = {}; in TestRemoveAttribute()
102 targeted_context.input_names = {}; in TestRemoveAttribute()
Drename_attribute_test.cc84 context.input_names = {}; in TestRenameAttribute()
106 targeted_context.input_names = {}; in TestRenameAttribute()
/external/tensorflow/tensorflow/contrib/predictor/
Dsaved_model_predictor.py93 input_names, argument
98 input_names_specified = input_names is not None
121 input_names=None, argument
150 signature_def_key, signature_def, input_names, output_names)
158 if input_names is None:
161 input_names = {k: v.name for k, v in signature_def.inputs.items()}
165 for k, v in input_names.items()}
Dsaved_model_predictor_test.py132 input_names = {'x': 'inputs/x:0',
137 input_names=input_names,
DREADME.md67 You can also specify input and output `Tensor`s by name using the `input_names`
73 input_names={'x': 'x:0', 'y': 'y:0'},
/external/tensorflow/tensorflow/core/kernels/
Dsave_op.cc56 static const char* input_names[3] = {"basename", "shard", "num_shards"}; in Compute() local
59 errors::InvalidArgument(input_names[i], in Compute()
79 static const char* input_names[2] = {"basename", "num_shards"}; in Compute() local
82 errors::InvalidArgument(input_names[i], in Compute()
Dremote_fused_graph_rewriter_transform.cc137 const std::vector<string>& inputs = context.input_names; in FuseRemoteGraph()
211 const std::vector<string>& inputs = context.input_names; in PlaceRemoteGraphArguments()
/external/tensorflow/tensorflow/contrib/meta_graph_transform/
Dmeta_graph_transform_test.py77 input_names = ['a', 'b']
83 base_meta_graph_def, input_names, output_names, transforms, tags)
106 input_names = ['i1', 'i2']
116 graph_def, input_names, output_names, initializer_names, transforms))
120 graph_def, input_names, output_names + init_nodes, transforms)
250 input_names = ['i1', 'i2']
289 graph_def, input_names, output_names, initializer_names,
292 gtt_mock.assert_called_once_with(graph_def, input_names, output_names,
309 input_names = ['i1', 'i2']
324 meta_graph_transform._do_transforms(graph_def, input_names,
[all …]
Dmeta_graph_transform.py70 def _gtt_transforms(graph_def, input_names, output_names, initializer_names, argument
95 return _graph_transforms.TransformGraph(graph_def, input_names,
185 def _sparsify_gather_transform(graph_def, input_names, output_names, argument
222 graph_def = _gtt_transforms(graph_def, input_names, output_names,
264 input_names, argument
302 transformed_graph_def, input_names, output_names,
311 transformed_graph_def, input_names, output_names,
315 transformed_graph_def, input_names, output_names,
322 transformed_graph_def, input_names, output_names,
706 base_meta_graph_def, input_names, output_names, transforms, tags, argument
[all …]
/external/tensorflow/tensorflow/python/client/
Dtf_session_helper.cc58 NameVector input_names; in TF_Run_wrapper_helper() local
75 input_names.push_back(key_string); in TF_Run_wrapper_helper()
97 TF_Run(session, run_options, input_names.data(), inputs_unsafe.data(), in TF_Run_wrapper_helper()
98 input_names.size(), const_cast<const char**>(output_names.data()), in TF_Run_wrapper_helper()
103 TF_PRun(session, handle, input_names.data(), inputs_unsafe.data(), in TF_Run_wrapper_helper()
104 input_names.size(), const_cast<const char**>(output_names.data()), in TF_Run_wrapper_helper()
161 const NameVector& input_names, in TF_PRunSetup_wrapper() argument
167 session, const_cast<const char**>(input_names.data()), input_names.size(), in TF_PRunSetup_wrapper()
/external/tensorflow/tensorflow/core/debug/
Ddebugger_state_impl.cc40 const int64 executor_step_index, const std::vector<string>& input_names, in PublishDebugMetadata() argument
44 executor_step_index, input_names, in PublishDebugMetadata()
Ddebugger_state_impl.h38 const std::vector<string>& input_names,
/external/tensorflow/tensorflow/core/kernels/hexagon/
Dhexagon_rewriter_transform.cc50 for (auto i = 0; static_cast<size_t>(i) < context.input_names.size(); ++i) { in RewriteQuantizedStrippedModelForHexagon()
51 const string& input_name = context.input_names.at(i); in RewriteQuantizedStrippedModelForHexagon()
/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_runner.cc133 std::vector<string> input_names; in Run() local
136 input_names.emplace_back(tensor_name); in Run()
148 graph_to_run.get(), input_names, output_names, {} /* target nodes */, in Run()
Ddebugger_state_interface.h53 const int64 executor_step_index, const std::vector<string>& input_names,
/external/tensorflow/tensorflow/tools/graph_transforms/python/
Dtransform_graph_test.py73 input_names = []
76 transformed_graph_def = TransformGraph(input_graph_def, input_names,
/external/tensorflow/tensorflow/contrib/lite/python/
Dop_hint.py285 input_names = [None] * len(call.inputs)
290 input_names[input_index] = _tensor_name_base(tensor)
296 current_graph_def, input_names, output_names, output_dtypes,
/external/vixl/tools/test_generator/
Dgenerator.py137 def __init__(self, name, seed, operand_names, input_names, operand_filter, argument
142 self.input_names = input_names
273 if input_type.name in self.input_names else [input_type.default]
Dparser.py298 input_names = json_test_case["inputs"] if "inputs" in json_test_case else []
315 return generator.TestCase(json_test_case["name"], seed, operand_names, input_names,
/external/autotest/client/cros/input_playback/
Dinput_playback.py265 input_names = glob.glob(os.path.join(device_dir, 'input', 'input*'))
266 for input_name in input_names:
273 logging.error('Input names found(%s): %s', device_dir, input_names)

123