| /external/tensorflow/tensorflow/core/common_runtime/ |
| debugger_state_interface.cc | 33 string tensor_name = local
38 oss << tensor_name << "|"; local
|
| kernel_benchmark_testlib.cc | 117 string tensor_name; local
118 TF_CHECK_OK(GetNodeAttr(node->attrs(), "tensor_name", &tensor_name));
123 recv_device, tensor_name, FrameAndIter(0, 0));
|
| graph_runner.cc | 135 const string& tensor_name = in.first; local
136 input_names.emplace_back(tensor_name);
138 tensor_name, FrameAndIter(0, 0));
|
| executor.cc | 101 string tensor_name; local
102 TF_CHECK_OK(GetNodeAttr(attrs, "tensor_name", &tensor_name));
106 "(", tensor_name, " @", recv_device);
109 string tensor_name; local
110 TF_CHECK_OK(GetNodeAttr(attrs, "tensor_name", &tensor_name));
114 "(", tensor_name, " @", send_device);
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| save_op_test.cc | 672 const string tensor_name = "my_tensor"; local
673 ops::Save(root, temp_filename, {tensor_name}, {{tensor}});
|
| sendrecv_ops.cc | 29 const string& tensor_name) {
32 recv_device, ";", tensor_name);
64 string tensor_name; local
65 OP_REQUIRES_OK(ctx, ctx->GetAttr("tensor_name", &tensor_name));
67 send_device_incarnation, tensor_name);
132 string tensor_name; local
133 OP_REQUIRES_OK(ctx, ctx->GetAttr("tensor_name", &tensor_name));
135 send_device_incarnation, tensor_name);
[all...] |
| restore_op_test.cc | 374 const string tensor_name = "tensor_int"; local
406 input_1.scalar<string>()() = tensor_name;
439 [&tensor_name](int x) -> string { return tensor_name; });
|
| save_restore_tensor.cc | 157 const string& tensor_name = tensor_name_t.flat<string>()(restore_index); local
176 context, reader->HasTensor(tensor_name, &saved_shape, &type),
177 errors::NotFound("Tensor name \"", tensor_name,
184 " instead: tensor_name = ", tensor_name));
215 reader->CopySliceData(tensor_name, slice_to_load, \
255 const string& tensor_name = tensor_names_flat(i); local
259 reader.LookupTensorShape(tensor_name, &restored_full_shape));
265 TF_RETURN_IF_ERROR(reader.Lookup(tensor_name, restored_tensor));
277 "tensor_name = ", tensor_name, "; shape in shape_and_slice spec "
[all...] |
| save_restore_v2_ops.cc | 113 const string& tensor_name = tensor_names_flat(i); variable
132 writer.AddSlice(tensor_name, shape, slice, tensor));
134 OP_REQUIRES_OK(context, writer.Add(tensor_name, tensor));
|
| debug_ops.h | 40 OP_REQUIRES_OK(context, context->GetAttr("tensor_name", &tensor_name_));
128 string tensor_name; local
130 OP_REQUIRES_OK(context, context->GetAttr("tensor_name", &tensor_name));
132 std::vector<string> name_items = str_util::Split(tensor_name, ':');
137 tensor_name, "\""));
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| graph_to_functiondef.cc | 211 const string tensor_name = strings::StrCat( local
213 tensor_renaming[strings::StrCat(node->name(), ":", i)] = tensor_name;
|
| /external/tensorflow/tensorflow/contrib/lite/toco/tflite/ |
| import.cc | 128 const string& tensor_name = local
130 model->CreateOptionalArray(tensor_name);
131 op->inputs.push_back(tensor_name);
|
| export.cc | 100 const string& tensor_name = array_pair.first; local
133 int index = tensors_map.at(tensor_name);
136 builder->CreateString(tensor_name), q_param);
|
| /external/tensorflow/tensorflow/contrib/graph_editor/ |
| subgraph.py | 462 def tensor_name(t): function in function:SubGraphView.__str__
479 print_list("inputs", self._input_ts, tensor_name)
480 print_list("outputs", self._output_ts, tensor_name)
|
| /external/tensorflow/tensorflow/contrib/session_bundle/ |
| bundle_shim_test.cc | 113 const string tensor_name = "foo_tensor"; local
118 tensor_name_to_dtype[tensor_name] = tensorflow::DT_STRING;
120 AddInputToSignatureDef(tensor_name, tensor_name_to_dtype, map_key,
123 EXPECT_EQ(tensor_name, signature_def.inputs().find(map_key)->second.name());
129 const string tensor_name = "foo_tensor"; local
134 tensor_name_to_dtype[tensor_name] = tensorflow::DT_STRING;
136 AddOutputToSignatureDef(tensor_name, tensor_name_to_dtype, map_key,
139 EXPECT_EQ(tensor_name, signature_def.outputs().find(map_key)->second.name());
|
| /external/tensorflow/tensorflow/core/debug/ |
| debug_graph_utils.cc | 78 string tensor_name = local
86 tensor_watches[tensor_name] = debug_ops;
87 tensor_tolerate_failures[tensor_name] =
94 tensor_watch_urls[tensor_name] = urls;
121 const string tensor_name = local
123 if (tensor_watches.find(tensor_name) == tensor_watches.end()) {
152 src_output_slot, src_dt, tensor_name, tensor_watches[tensor_name],
153 tensor_watch_urls[tensor_name], ©_node);
158 tensor_name, ", due to: ", copy_s.error_message()))
[all...] |
| /external/tensorflow/tensorflow/core/grappler/ |
| grappler_item_builder_test.cc | 136 Output tensor_name = local
139 tensor_name, DataType::DT_FLOAT);
197 Output tensor_name = local
200 tensor_name, DataType::DT_FLOAT);
202 tensor_name, DataType::DT_FLOAT);
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| message_wrappers.cc | 739 const string& InMemoryRunStepResponse::tensor_name(size_t i) const { function in class:tensorflow::InMemoryRunStepResponse
784 const string& OwnedProtoRunStepResponse::tensor_name(size_t i) const { function in class:tensorflow::OwnedProtoRunStepResponse
836 const string& NonOwnedProtoRunStepResponse::tensor_name(size_t i) const { function in class:tensorflow::NonOwnedProtoRunStepResponse
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| graph_properties_test.cc | 638 Output tensor_name = local
640 Output restore = ops::Restore(s.WithOpName("restore"), filename, tensor_name,
647 ops::RestoreSlice(s.WithOpName("restore_slice"), filename, tensor_name,
653 ops::RestoreSlice(s.WithOpName("restore_v2"), filename, tensor_name,
697 Output tensor_name = local
699 Output restore = ops::Restore(s.WithOpName("restore"), filename, tensor_name,
[all...] |
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| hexagon_control_wrapper.cc | 326 const string tensor_name = AddPort(node_name); local
327 CHECK(input_port_map_.count(tensor_name) > 0);
328 const int port = input_port_map_.at(tensor_name);
385 const string tensor_name = AddPort(node_name); local
386 CHECK(output_port_map_.count(tensor_name) > 0);
387 const int port = output_port_map_.at(tensor_name);
|
| /external/tensorflow/tensorflow/python/debug/lib/ |
| debug_data.py | 417 def tensor_name(self): member in class:DebugTensorDatum
431 (`str`) A watch key, in the form of `tensor_name`:`debug_op`.
[all...] |
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| memory_optimizer.cc | 507 string tensor_name = local
509 addn_list[tensor_name].insert(&node);
542 string tensor_name = strings::StrCat(live.node, ":", live.output_id); local
543 auto it = addn_list.find(tensor_name);
[all...] |
