| /external/tensorflow/tensorflow/python/util/ |
| kernel_registry.cc | 17 #include "tensorflow/core/framework/node_def.pb.h"
29 tensorflow::NodeDef node_def; local
30 if (!node_def.ParseFromString(serialized_node_def)) {
31 LOG(WARNING) << "Error parsing node_def";
37 tensorflow::OpRegistry::Global()->LookUp(node_def.op(), &op_reg_data);
39 LOG(WARNING) << "Op " << node_def.op() << " not found: " << status;
42 AddDefaultsToNodeDef(op_reg_data->op_def, &node_def);
45 if (!tensorflow::DeviceNameUtils::ParseFullName(node_def.device(),
47 LOG(WARNING) << "Failed to parse device from node_def: "
48 << node_def.ShortDebugString()
53 node_def, nullptr \/* kernel_def *\/, &class_name) local
[all...] |
| /external/tensorflow/tensorflow/core/util/ |
| mirror_pad_mode.cc | 24 Status GetNodeAttr(const NodeDef& node_def, StringPiece attr_name,
27 TF_RETURN_IF_ERROR(GetNodeAttr(node_def, attr_name, &str_value));
|
| padding.cc | 24 Status GetNodeAttr(const NodeDef& node_def, StringPiece attr_name,
27 TF_RETURN_IF_ERROR(GetNodeAttr(node_def, attr_name, &str_value));
|
| mirror_pad_mode.h | 47 Status GetNodeAttr(const NodeDef& node_def, StringPiece attr_name,
|
| padding.h | 48 Status GetNodeAttr(const NodeDef& node_def, StringPiece attr_name,
|
| /external/tensorflow/tensorflow/core/framework/ |
| node_def_util.h | 51 string SummarizeNodeDef(const NodeDef& node_def);
55 // Adds an attr with name <name> and value <value> to *node_def.
57 void AddNodeAttr(StringPiece name, const AttrValue& value, NodeDef* node_def);
58 void AddNodeAttr(StringPiece name, StringPiece value, NodeDef* node_def);
59 void AddNodeAttr(StringPiece name, const char* value, NodeDef* node_def);
60 void AddNodeAttr(StringPiece name, int32 value, NodeDef* node_def);
61 void AddNodeAttr(StringPiece name, int64 value, NodeDef* node_def);
62 void AddNodeAttr(StringPiece name, float value, NodeDef* node_def);
63 void AddNodeAttr(StringPiece name, double value, NodeDef* node_def);
64 void AddNodeAttr(StringPiece name, bool value, NodeDef* node_def);
[all...] |
| node_def_util_test.cc | 38 NodeDef node_def; local
39 EXPECT_TRUE(protobuf::TextFormat::MergeFromString(text, &node_def));
40 return node_def;
44 NodeDef node_def; local
45 TF_EXPECT_OK(builder.Finalize(&node_def));
46 return node_def;
76 const NodeDef node_def = ToNodeDef(R"proto( local
79 ExpectSuccess(node_def, op);
81 EXPECT_EQ("n = In[T=DT_FLOAT](a)", SummarizeNodeDef(node_def));
84 NodeDef bad = node_def;
141 const NodeDef node_def = ToNodeDef(R"proto( local
161 const NodeDef node_def = ToNodeDef(R"proto( local
187 const NodeDef node_def = ToNodeDef(R"proto( local
216 const NodeDef node_def = ToNodeDef(R"proto( local
274 const NodeDef node_def = ToNodeDef(R"proto( local
375 const NodeDef node_def = ToNodeDef( local
[all...] |
| op_compatibility_test.cc | 45 TF_CHECK_OK(OpRegistry::Global()->LookUpOpDef(node_def()->op(), &op_def));
50 // Record the original signature before we change *node_def().
52 TF_ASSERT_OK(InOutTypesForNode(*node_def(), old_op_def, &old_in_types,
57 AddDefaultsToNodeDef(*new_op_def, node_def());
60 TF_ASSERT_OK(ValidateNodeDef(*node_def(), *new_op_def));
62 TF_ASSERT_OK(InOutTypesForNode(*node_def(), *new_op_def, &new_in_types,
93 // Test OpDefCompatible gives the same answer without the node_def.
106 // Record the original signature before we change *node_def().
108 TF_ASSERT_OK(InOutTypesForNode(*node_def(), old_op_def, &old_in_types,
113 AddDefaultsToNodeDef(*new_op_def, node_def());
[all...] |
| graph_def_util.cc | 26 #include "tensorflow/core/framework/node_def.pb.h"
64 NodeDef* node_def = graph_def->mutable_node(i); local
66 TF_RETURN_IF_ERROR(op_registry.LookUpOpDef(node_def->op(), &op_def));
67 AddDefaultsToNodeDef(*op_def, node_def);
74 NodeDef* node_def, const OpRegistryInterface& consumer_op_registry,
80 producer_op_registry.LookUpOpDef(node_def->op(), &producer_op_def));
82 consumer_op_registry.LookUpOpDef(node_def->op(), &consumer_op_def));
85 for (const auto& attr : node_def->attr()) {
95 " but found in node: ", SummarizeNodeDef(*node_def));
109 node_def->mutable_attr()->erase(attr_name)
132 NodeDef* node_def = graph_def->mutable_node(n); local
142 NodeDef* node_def = func_def->mutable_node_def(n); local
[all...] |
| node_def_util.cc | 24 #include "tensorflow/core/framework/node_def.pb.h"
47 AttrSlice::AttrSlice(const NodeDef& node_def)
48 : ndef_(&node_def), attrs_(&ndef_->attr()) {}
87 string SummarizeNodeDef(const NodeDef& node_def) {
88 string ret = strings::StrCat(node_def.name(), " = ", node_def.op(), "[");
89 strings::StrAppend(&ret, SummarizeAttrsHelper(node_def, node_def.device()));
94 for (const string& input : node_def.input()) {
246 bool HasNodeAttr(const NodeDef& node_def, StringPiece attr_name)
[all...] |
| fake_input.cc | 31 FakeInputImpl(const OpDef* op_def, int in_index, const NodeDef* node_def,
60 const NodeDef* node_def, NodeDefBuilder* builder)
64 node_def_(node_def),
193 return [](const OpDef& op_def, int in_index, const NodeDef& node_def,
195 FakeInputImpl impl(&op_def, in_index, &node_def, builder);
201 return [dt](const OpDef& op_def, int in_index, const NodeDef& node_def,
203 FakeInputImpl impl(&op_def, in_index, &node_def, builder);
210 return [n](const OpDef& op_def, int in_index, const NodeDef& node_def,
212 FakeInputImpl impl(&op_def, in_index, &node_def, builder);
219 return [n, dt](const OpDef& op_def, int in_index, const NodeDef& node_def,
[all...] |
| memory_types_test.cc | 56 NodeDef node_def; local
63 .Finalize(&node_def));
66 TF_EXPECT_OK(MemoryTypesForNode(OpRegistry::Global(), DEVICE_CPU, node_def,
80 TF_EXPECT_OK(MemoryTypesForNode(OpRegistry::Global(), DEVICE_GPU, node_def,
|
| /external/tensorflow/tensorflow/core/ops/ |
| state_ops_test.cc | 31 .Finalize(&op.node_def));
46 .Finalize(&op.node_def));
56 .Finalize(&op.node_def));
68 .Finalize(&op.node_def));
78 .Finalize(&op.node_def));
85 .Finalize(&op.node_def));
91 .Finalize(&op.node_def));
101 .Finalize(&op.node_def));
107 .Finalize(&op.node_def));
113 .Finalize(&op.node_def));
[all...] |
| candidate_sampling_ops_test.cc | 38 .Finalize(&op.node_def));
58 .Finalize(&op.node_def));
|
| string_ops_test.cc | 34 .Finalize(&op.node_def));
|
| /external/tensorflow/tensorflow/python/framework/ |
| errors_impl.py | 39 def __init__(self, node_def, op, message, error_code):
43 node_def: The `node_def_pb2.NodeDef` proto representing the op that
51 self._node_def = node_def
68 instead use the @{tf.OpError.node_def} to
82 def node_def(self): member in class:OpError
194 def __init__(self, node_def, op, message):
196 super(CancelledError, self).__init__(node_def, op, message, CANCELLED)
213 def __init__(self, node_def, op, message, error_code=UNKNOWN):
215 super(UnknownError, self).__init__(node_def, op, message, error_code)
233 def __init__(self, node_def, op, message)
[all...] |
| op_def_library_test.py | 123 """, out.op.node_def)
128 """, out.op.node_def)
133 """, out.op.node_def)
138 """, out.op.node_def)
184 """, op.node_def)
193 """, out.op.node_def)
200 """, out.op.node_def)
207 """, out.op.node_def)
221 """, out.op.node_def)
228 """, out.op.node_def)
[all...] |
| /external/tensorflow/tensorflow/core/graph/ |
| graph_constructor.cc | 28 #include "tensorflow/core/framework/node_def.pb.h"
46 inline bool IsMerge(const NodeDef& node_def) {
47 return node_def.op() == "Merge" || node_def.op() == "RefMerge";
50 inline bool IsNextIteration(const NodeDef& node_def) {
51 return node_def.op() == "NextIteration" ||
52 node_def.op() == "RefNextIteration";
186 Status IsNodeFullyMapped(const NodeDef& node_def, bool* is_node_mapped);
187 Status ValidateColocationConstraints(const NodeDef& node_def);
188 Status MakeNode(const NodeDef& node_def, Node** node)
414 const NodeDef& node_def = *node_defs_[n]; local
458 const NodeDef& node_def = *node_defs[n]; local
476 const NodeDef& node_def = *node_defs_[n]; local
905 const NodeDef* node_def; local
1133 const NodeDef* node_def = node_defs_[pair->second.gdef_index]; local
[all...] |
| validate.cc | 19 #include "tensorflow/core/framework/node_def.pb.h"
33 for (const NodeDef& node_def : graph_def.node()) {
34 // Look up the OpDef for the node_def's op name.
36 TF_RETURN_IF_ERROR(op_registry.LookUpOpDef(node_def.op(), &op_def));
37 TF_RETURN_IF_ERROR(ValidateNodeDef(node_def, *op_def));
|
| /external/tensorflow/tensorflow/core/kernels/ |
| ops_testutil_test.cc | 30 .Finalize(node_def()));
41 .Finalize(node_def()));
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| graph_to_functiondef.cc | 23 #include "tensorflow/core/framework/node_def.pb.h"
161 NodeDef* node_def = fdef->add_node_def(); local
162 *node_def = node->def();
164 node_def->set_device(node->assigned_device_name());
166 node_def->set_name(node_names.Uniquify(node->name()));
169 node_def->clear_input();
196 node_def->add_input(
202 node_def->add_input(strings::StrCat("^", edge->src()->name()));
212 node_def->name(), ":", output.first, ":", i - output.second.first);
230 NodeDef* node_def = fdef->mutable_node_def(n_index) local
[all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| sharding_util.cc | 17 #include "tensorflow/core/framework/node_def.pb.h"
30 GetShardingFromNodeDef(const NodeDef& node_def) {
31 if (!HasNodeAttr(node_def, kShardingAttribute)) {
36 TF_RETURN_IF_ERROR(GetNodeAttr(node_def, kShardingAttribute, &value));
82 ParseShardingFromDevice(const NodeDef& node_def, int num_cores_per_replica) {
83 const string& device_name = node_def.device();
85 GetShardingFromNodeDef(node_def));
|
| /external/tensorflow/tensorflow/python/tools/ |
| selective_registration_header_lib.py | 51 for node_def in graph_def.node:
52 if not node_def.device:
53 node_def.device = '/cpu:0'
55 node_def.SerializeToString())
57 op_and_kernel = (str(node_def.op), str(kernel_class.decode('utf-8')))
62 'Warning: no kernel found for op %s' % node_def.op, file=sys.stderr)
|
| /external/tensorflow/tensorflow/contrib/hvx/hvx_ops_support_checker/ |
| hvx_ops_support_checker_main.cc | 24 #include "tensorflow/core/framework/node_def.pb.h"
57 static void SummarizeNode(const NodeDef& node_def,
59 LOG(INFO) << "Node(" << node_def.name() << ")";
60 LOG(INFO) << " op: " << node_def.op();
61 for (const string& input : node_def.input()) {
67 node_def, &data_types, &shapes);
78 static void DumpRemoteFusedGraph(const NodeDef& node_def) {
82 GetNodeAttr(node_def,
88 LOG(INFO) << "Node name: " << node_def.name();
|
| /external/tensorflow/tensorflow/contrib/receptive_field/python/util/ |
| graph_compute_order.py | 41 for node_def in graph_def.node:
42 name_to_node[node_def.name] = node_def
103 node_def = name_to_node[current]
109 node_info[current] = _node_info(order, node_def, input_size, output_size)
117 for each in node_def.input:
142 node_def, name_to_node, input_size, force=True))
155 node_info[current] = _node_info(order, node_def, input_size, output_size)
|
