Home
last modified time | relevance | path

Searched refs:tensor_name (Results 1 – 25 of 124) sorted by relevance

12345

/external/tensorflow/tensorflow/core/debug/
Ddebug_graph_utils.cc78 string tensor_name = in InsertNodes() local
86 tensor_watches[tensor_name] = debug_ops; in InsertNodes()
87 tensor_tolerate_failures[tensor_name] = in InsertNodes()
94 tensor_watch_urls[tensor_name] = urls; in InsertNodes()
121 const string tensor_name = in InsertNodes() local
123 if (tensor_watches.find(tensor_name) == tensor_watches.end()) { in InsertNodes()
152 src_output_slot, src_dt, tensor_name, tensor_watches[tensor_name], in InsertNodes()
153 tensor_watch_urls[tensor_name], &copy_node); in InsertNodes()
158 tensor_name, ", due to: ", copy_s.error_message())); in InsertNodes()
166 for (size_t i = 0; i < tensor_watches[tensor_name].size(); ++i) { in InsertNodes()
[all …]
Ddebug_graph_utils.h89 static const string GetDebugNodeName(const string& tensor_name,
97 const string& tensor_name, const std::vector<string>& debug_ops,
116 const string& tensor_name,
/external/tensorflow/tensorflow/contrib/session_bundle/
Dbundle_shim.cc54 internal::AddInputToSignatureDef(regression_signature.input().tensor_name(), in BuildRegressionSignatureDef()
57 internal::AddOutputToSignatureDef(regression_signature.output().tensor_name(), in BuildRegressionSignatureDef()
69 classification_signature.input().tensor_name(), tensor_name_to_dtype, in BuildClassificationSignatureDef()
72 classification_signature.classes().tensor_name(), tensor_name_to_dtype, in BuildClassificationSignatureDef()
75 classification_signature.scores().tensor_name(), tensor_name_to_dtype, in BuildClassificationSignatureDef()
106 internal::AddInputToSignatureDef(map_entry.second.tensor_name(), in MaybeBuildPredictSignatureDef()
113 internal::AddOutputToSignatureDef(map_entry.second.tensor_name(), in MaybeBuildPredictSignatureDef()
244 const string& tensor_name, in AddInputToSignatureDef() argument
247 if (tensor_name.empty()) { in AddInputToSignatureDef()
254 std::size_t pos = tensor_name.find(":"); in AddInputToSignatureDef()
[all …]
Dbundle_shim.py36 def _add_input_to_signature_def(tensor_name, map_key, signature_def): argument
48 tensor_info = meta_graph_pb2.TensorInfo(name=tensor_name)
52 def _add_output_to_signature_def(tensor_name, map_key, signature_def): argument
65 tensor_info = meta_graph_pb2.TensorInfo(name=tensor_name)
89 _add_input_to_signature_def(regression_signature.input.tensor_name,
92 _add_output_to_signature_def(regression_signature.output.tensor_name,
99 _add_input_to_signature_def(classification_signature.input.tensor_name,
102 _add_output_to_signature_def(classification_signature.classes.tensor_name,
105 _add_output_to_signature_def(classification_signature.scores.tensor_name,
148 _add_input_to_signature_def(val.tensor_name, key, signature_def)
[all …]
Dexporter.py88 signature.regression_signature.input.tensor_name = input_tensor.name
89 signature.regression_signature.output.tensor_name = output_tensor.name
109 signature.classification_signature.input.tensor_name = input_tensor.name
111 signature.classification_signature.classes.tensor_name = classes_tensor.name
113 signature.classification_signature.scores.tensor_name = scores_tensor.name
130 signature.generic_signature.map[name].tensor_name = tensor.name
228 asset.tensor_binding.tensor_name = tensor.name
Dsignature.cc120 output_tensor_names.push_back(signature.classes().tensor_name()); in RunClassification()
123 output_tensor_names.push_back(signature.scores().tensor_name()); in RunClassification()
128 session->Run({{signature.input().tensor_name(), input}}, in RunClassification()
175 output_tensor_names.push_back(signature.output().tensor_name()); in RunRegression()
180 session->Run({{signature.input().tensor_name(), regression_input}}, in RunRegression()
254 bound_inputs->push_back({mapped->second.tensor_name(), entry.second}); in BindGenericInputs()
271 bound_names->push_back(mapped->second.tensor_name()); in BindGenericNames()
Dexporter_test.py167 default_signature.classification_signature.input.tensor_name, "v0:0")
169 self.assertEquals(bindings["logical_input_A"].tensor_name, "v0:0")
170 self.assertEquals(bindings["logical_input_B"].tensor_name, "v1:0")
173 self.assertEquals(read_foo_signature.input.tensor_name, "v0:0")
174 self.assertEquals(read_foo_signature.output.tensor_name, "v1:0")
188 self.assertEquals("filename42:0", asset.tensor_binding.tensor_name)
/external/tensorflow/tensorflow/python/debug/lib/
Dstepper.py370 tensor_name = node_name + ":%d" % i
371 if tensor_name in self._closure_elements:
392 def override_tensor(self, tensor_name, overriding_val): argument
404 if not isinstance(tensor_name, six.string_types):
405 raise TypeError("Expected type str; got type %s" % type(tensor_name))
407 node_name = self._get_node_name(tensor_name)
412 (tensor_name, repr(self._fetch_names)))
414 self._override_tensors[tensor_name] = overriding_val
417 self._invalidate_transitively_outgoing_cache(tensor_name)
419 def remove_override(self, tensor_name): argument
[all …]
Dsource_utils.py230 tensor_name = output_tensor.name
231 path_to_tensor_names[file_path].add(tensor_name)
236 tensor_name = "%s:%s" % (node_name, output_slot)
237 if tensor_name not in tensor_name_to_num_dumps:
238 tensor_name_to_num_dumps[tensor_name] = len(
244 tensor_name_to_num_dumps.get(tensor_name, 0)
245 for tensor_name in path_to_tensor_names[path])
/external/tensorflow/tensorflow/core/kernels/
Dsave_restore_tensor.cc159 const string& tensor_name = tensor_name_t.flat<string>()(restore_index); in RestoreTensor() local
181 context, reader->HasTensor(tensor_name, &saved_shape, &type), in RestoreTensor()
182 errors::NotFound("Tensor name \"", tensor_name, in RestoreTensor()
189 " instead: tensor_name = ", tensor_name)); in RestoreTensor()
220 reader->CopySliceData(tensor_name, slice_to_load, \ in RestoreTensor()
250 if (!reader->LookupTensorShape(tensor_name, &restored_full_shape).ok()) { in should_run_in_pool()
271 reader->LookupTensorShape(tensor_name, &restored_full_shape)); in run()
273 VLOG(1) << "Restoring tensor " << idx << " : " << tensor_name << " : " in run()
280 TF_RETURN_IF_ERROR(reader->Lookup(tensor_name, restored_tensor)); in run()
293 "tensor_name = ", tensor_name, "; shape in shape_and_slice spec ", in run()
[all …]
Dsendrecv_ops.cc29 const string& tensor_name) { in GetRendezvousKeyPrefix() argument
32 recv_device, ";", tensor_name); in GetRendezvousKeyPrefix()
64 string tensor_name; in SendOp() local
65 OP_REQUIRES_OK(ctx, ctx->GetAttr("tensor_name", &tensor_name)); in SendOp()
67 send_device_incarnation, tensor_name); in SendOp()
132 string tensor_name; in RecvOp() local
133 OP_REQUIRES_OK(ctx, ctx->GetAttr("tensor_name", &tensor_name)); in RecvOp()
135 send_device_incarnation, tensor_name); in RecvOp()
/external/tensorflow/tensorflow/python/debug/cli/
Dstepper_cli.py349 tensor_name.split(":")[0] for tensor_name in intermediate_tensor_names]
528 tensor_name, tensor_slicing = (
529 command_parser.parse_tensor_name_with_slicing(parsed.tensor_name))
531 tensor_names = self._resolve_tensor_names(tensor_name)
534 self._MESSAGE_TEMPLATES["NOT_IN_CLOSURE"] % tensor_name)
537 self._MESSAGE_TEMPLATES["MULTIPLE_TENSORS"] % tensor_name)
539 tensor_name = tensor_names[0]
542 tensor_value = self._node_stepper.get_tensor_value(tensor_name)
548 tensor_name,
574 tensor_names = self._resolve_tensor_names(parsed.tensor_name)
[all …]
Devaluator.py134 tensor_name = match.group(0)[1:-1].strip()
136 _parse_debug_tensor_name(tensor_name))
137 if tensor_name not in self._cached_tensor_values:
146 self._cached_tensor_values[tensor_name] = value
149 "self._cached_tensor_values['" + tensor_name + "']" +
Danalyzer_cli_test.py253 for tensor_name, op_type in zip(expected_tensor_names, expected_op_types):
254 tst.assertIn(tensor_name, tensor_names)
255 index = tensor_names.index(tensor_name)
1000 tensor_name = node_name + ":0"
1001 out = self._registry.dispatch_command("node_info", [tensor_name])
1026 tensor_name = node_name + ":0"
1028 "print_tensor", [tensor_name], screen_info={"cols": 80})
1031 "Tensor \"%s:DebugIdentity\":" % tensor_name,
1052 tensor_name = node_name + ":0"
1055 "print_tensor", [tensor_name, "-w", npy_path],
[all …]
/external/tensorflow/tensorflow/contrib/learn/python/learn/utils/
Dinspect_checkpoint.py30 def print_tensors_in_checkpoint_file(file_name, tensor_name): argument
43 if not tensor_name:
48 print("tensor_name: ", tensor_name)
49 print(checkpoint_utils.load_variable(file_name, tensor_name))
63 print_tensors_in_checkpoint_file(FLAGS.file_name, FLAGS.tensor_name)
/external/tensorflow/tensorflow/python/tools/
Dinspect_checkpoint.py32 def print_tensors_in_checkpoint_file(file_name, tensor_name, all_tensors, argument
55 elif not tensor_name:
58 print("tensor_name: ", tensor_name)
59 print(reader.get_tensor(tensor_name))
117 print_tensors_in_checkpoint_file(FLAGS.file_name, FLAGS.tensor_name,
/external/tensorflow/tensorflow/contrib/meta_graph_transform/
Dmeta_graph_transform.py45 def _op_name(tensor_name): argument
48 if tensor_name[0] == '^':
49 tensor_name = tensor_name[1:]
50 if ':' in tensor_name:
51 op_name, _ = tensor_name.split(':')
53 return tensor_name
635 def _is_removed(tensor_name, removed_op_names): argument
638 if tensor_name.split(':')[0] == removed_op_name:
672 def _check_tensor_not_removed(tensor_name, removed_op_names): argument
682 if not tensor_name:
[all …]
/external/tensorflow/tensorflow/core/ops/
Ddebug_ops.cc62 .Attr("tensor_name: string = ''")
78 tensor_name: The name of the input tensor.
91 .Attr("tensor_name: string = ''")
103 tensor_name: Name of the input tensor.
119 .Attr("tensor_name: string = ''")
131 tensor_name: Name of the input tensor.
147 .Attr("tensor_name: string = ''")
/external/tensorflow/tensorflow/core/common_runtime/
Dmemory_types.cc121 static Node* Send(Graph* g, const string& tensor_name, in Send() argument
126 .Attr("tensor_name", tensor_name) in Send()
135 static Node* Recv(Graph* g, const string& tensor_name, in Recv() argument
141 .Attr("tensor_name", tensor_name) in Recv()
186 const string tensor_name = GetTensorName(e); in EnsureMemoryTypes() local
188 Send(g, tensor_name, device_name, (item.sm == HOST_MEMORY), e); in EnsureMemoryTypes()
189 recv = Recv(g, tensor_name, device_name, (item.dm == HOST_MEMORY), e); in EnsureMemoryTypes()
/external/tensorflow/tensorflow/contrib/framework/python/framework/
Dcheckpoint_utils.py107 def _set_checkpoint_initializer(variable, file_pattern, tensor_name, slice_spec, argument
120 file_pattern, [tensor_name], [slice_spec], [base_type], name=name)[0]
125 tensor_name): argument
135 _set_checkpoint_initializer(v, file_pattern, tensor_name,
138 _set_checkpoint_initializer(variable_or_list, file_pattern, tensor_name, "")
/external/tensorflow/tensorflow/contrib/model_pruning/python/
Dstrip_pruning_vars_lib.py37 def _node_name(tensor_name): argument
39 if ':' not in tensor_name:
40 return tensor_name
42 return tensor_name.split(':')[0]
/external/tensorflow/tensorflow/core/distributed_runtime/
Dworker_cache_logger.cc87 const string& tensor_name, in RecordRecvTensor() argument
91 RecordDataTransfer(step_id, start_usecs, end_usecs, tensor_name, src_device, in RecordRecvTensor()
97 const string& tensor_name, in RecordDataTransfer() argument
114 auto label = strings::StrCat(byte_string, rate_string, tensor_name, in RecordDataTransfer()
/external/tensorflow/tensorflow/python/tpu/
Dtensor_tracer.py923 def _make_tensor_trace_fun(self, tensor_name): argument
936 def _print_tensor(tensor_name, num_elements, tensor, output_tensor): argument
954 if tensor_name not in self._tensorname_idx_map:
956 'Tensor name %s is not in the tensorname_idx_map'%tensor_name)
957 msg = '%d'%self._tensorname_idx_map[tensor_name]
959 msg = '"%s"'%tensor_name
975 return _print_tensor(tensor_name, self._part_tensor_size,
981 return _print_tensor(tensor_name, -1, tensor, tensor)
1154 tensor_name = out_tensor.name
1155 if tensor_name in self._traced_tensorname_to_cache_idx_map:
[all …]
/external/tensorflow/tensorflow/python/training/
Dcheckpoint_utils.py291 tensor_name, argument
313 ckpt_file, [tensor_name], [slice_spec], [base_type], name=name)[0]
332 tensor_name): argument
357 _set_checkpoint_initializer(v, ckpt_file, tensor_name, slice_info.spec)
359 _set_checkpoint_initializer(variable_or_list, ckpt_file, tensor_name, "")
/external/tensorflow/tensorflow/lite/experimental/micro/
Dsimple_tensor_allocator.cc114 const char* tensor_name = flatbuffer_tensor.name()->c_str(); in AllocateTensor() local
115 if (tensor_name == nullptr) { in AllocateTensor()
116 tensor_name = "<None>"; in AllocateTensor()
121 tensor_name, result->bytes, (data_size_max_ - data_size_)); in AllocateTensor()

12345