| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | fold_constants_lib.cc | 121 std::unordered_set<string> input_names; in RewriteInputsAsPlaceholders() local
122 for (const string& input_name : context.input_names) { in RewriteInputsAsPlaceholders()
123 input_names.emplace(ParseTensorName(input_name).first); in RewriteInputsAsPlaceholders()
127 if (input_names.find(node.name()) == input_names.end()) { in RewriteInputsAsPlaceholders()
152 std::unordered_set<TensorId, TensorId::Hasher> input_names; in RemoveUnusedNodes() local
153 for (const string& input : context.input_names) { in RemoveUnusedNodes()
154 input_names.insert(ParseTensorName(input)); in RemoveUnusedNodes()
175 if (input_names.count(id) > 0) { in RemoveUnusedNodes()
185 for (const TensorId& id : input_names) { in RemoveUnusedNodes()
242 std::vector<TensorId> input_names; in FoldConstants() local
[all …]
|
| D | remove_nodes_test.cc | 94 context.input_names = {}; in TestRemoveNodes()
149 context.input_names = {}; in TestRemoveOutputNodes()
198 context.input_names = {}; in TestRemoveChainedNodes()
248 context.input_names = {}; in TestRemoveMultipleInputs()
|
| D | backports_test.cc | 85 context.input_names = {}; in TestBackportConcatV2()
157 context.input_names = {}; in TEST()
188 context.input_names = {}; in TEST()
|
| D | insert_logging_test.cc | 70 context.input_names = {}; in TestInsertLogging()
117 context.input_names = {}; in TestInsertLoggingByOpType()
166 context.input_names = {}; in TestInsertLoggingByPrefix()
|
| D | set_device_test.cc | 86 context.input_names = {}; in TEST()
107 context.input_names = {}; in TEST()
|
| D | remove_attribute_test.cc | 84 context.input_names = {}; in TestRemoveAttribute()
102 targeted_context.input_names = {}; in TestRemoveAttribute()
|
| /external/tensorflow/tensorflow/contrib/predictor/ |
| D | saved_model_predictor.py | 88 input_names, argument
93 input_names_specified = input_names is not None
116 input_names=None, argument
147 signature_def_key, signature_def, input_names, output_names)
155 if input_names is None:
158 input_names = {k: v.name for k, v in signature_def.inputs.items()}
162 for k, v in input_names.items()}
|
| D | predictor_factories.py | 111 input_names=None, argument
149 input_names=input_names,
|
| D | saved_model_predictor_test.py | 132 input_names = {'x': 'inputs/x:0',
137 input_names=input_names,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | save_op.cc | 56 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()
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| D | rank_two_test.py | 35 input_names = ["input", "input2"]
44 dtype=dtypes.float32, shape=input_dims[i], name=input_names[i])
65 input_names=input_names,
|
| D | base_test.py | 69 input_names=[input_name],
126 input_names=[input_name],
178 input_names=[input_name],
231 input_names=[input_name],
264 input_names=[input_name],
298 input_names=[input_name],
347 input_names=[input_name],
|
| D | conv2d_test.py | 106 input_names=["input"],
131 input_names=["input"],
181 input_names=[input_name],
222 input_names=[input_name],
|
| D | tf_trt_integration_test_base.py | 247 return [input_name + ":0" for input_name in params.input_names]
271 assert len(params.input_names) == len(data)
303 nodes_blacklist=params.input_names + params.output_names,
480 if self._ToString(node.name) in params.input_names:
484 assert len(params.input_names) == len(input_dtypes)
489 for i in range(len(params.input_names)):
490 dtype = input_dtypes[params.input_names[i]]
|
| D | topk_test.py | 47 input_names=[input_name],
78 input_names=[input_name],
|
| /external/tensorflow/tensorflow/contrib/meta_graph_transform/ |
| D | meta_graph_transform_test.py | 77 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 …]
|
| D | meta_graph_transform.py | 74 def _gtt_transforms(graph_def, input_names, output_names, initializer_names, argument
99 return _graph_transforms.TransformGraph(graph_def, input_names,
189 def _sparsify_gather_transform(graph_def, input_names, output_names, argument
226 graph_def = _gtt_transforms(graph_def, input_names, output_names,
268 input_names, argument
306 transformed_graph_def, input_names, output_names,
315 transformed_graph_def, input_names, output_names,
319 transformed_graph_def, input_names, output_names,
326 transformed_graph_def, input_names, output_names,
716 base_meta_graph_def, input_names, output_names, transforms, tags, argument
[all …]
|
| /external/tensorflow/tensorflow/python/keras/saving/ |
| D | saving_utils.py | 68 input_names = model.input_names
76 flat_input_names = nest.flatten(input_names)
|
| /external/tensorflow/tensorflow/python/client/ |
| D | tf_session_helper.cc | 72 NameVector input_names; in TF_Run_wrapper_helper() local
89 input_names.push_back(key_string); in TF_Run_wrapper_helper()
111 TF_Run(session, run_options, input_names.data(), inputs_unsafe.data(), in TF_Run_wrapper_helper()
112 input_names.size(), const_cast<const char**>(output_names.data()), in TF_Run_wrapper_helper()
117 TF_PRun(session, handle, input_names.data(), inputs_unsafe.data(), in TF_Run_wrapper_helper()
118 input_names.size(), const_cast<const char**>(output_names.data()), in TF_Run_wrapper_helper()
325 const NameVector& input_names, in TF_PRunSetup_wrapper() argument
331 session, const_cast<const char**>(input_names.data()), input_names.size(), in TF_PRunSetup_wrapper()
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | load_v1_in_v2.py | 101 input_names, input_specs = zip(*signature_def.inputs.items())
109 signature_fn._arg_keywords = input_names
110 if len(input_names) == 1:
|
| /external/tensorflow/tensorflow/core/debug/ |
| D | debugger_state_impl.cc | 43 const int64 executor_step_index, const std::vector<string>& input_names, in PublishDebugMetadata() argument
47 executor_step_index, input_names, in PublishDebugMetadata()
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | hexagon_rewriter_transform.cc | 50 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/ |
| D | graph_runner.cc | 131 std::vector<string> input_names; in Run() local
134 input_names.emplace_back(tensor_name); in Run()
146 graph_to_run.get(), input_names, output_names, {} /* target nodes */, in Run()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/python/ |
| D | transform_graph_test.py | 73 input_names = []
76 transformed_graph_def = TransformGraph(input_graph_def, input_names,
|
| /external/vixl/tools/test_generator/ |
| D | generator.py | 137 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]
|