| /external/tensorflow/tensorflow/core/framework/ |
| node_def_builder.cc | 41 node_def_.set_name(name.ToString());
53 node_def_.set_name(name.ToString());
59 node_def_.set_op(op_def_->name());
81 Status status = fake_input(*op_def_, inputs_specified_, node_def_, this);
171 node_def_.add_input(strings::StrCat(src_node, ":", src_index));
173 node_def_.add_input(src_node.ToString());
201 node_def_.set_device(device_spec.ToString());
223 node_def_.name(), "'");
226 (*errors_ptr)[0], " while building NodeDef '", node_def_.name(),
231 node_def_.name(), "' using ", SummarizeOpDef(*op_def_), ":\n"
[all...] |
| fake_input.cc | 48 const NodeDef* const node_def_; member in class:tensorflow::__anon39560::FakeInputImpl
64 node_def_(node_def),
104 Status status = GetNodeAttr(*node_def_, arg_->type_list_attr(), &dts);
131 Status status = GetNodeAttr(*node_def_, arg_->number_attr(), n);
148 Status status = GetNodeAttr(*node_def_, arg_->type_attr(), dt);
|
| node_def_builder.h | 133 const string& node_name() const { return node_def_.name(); }
154 // Add "src_node:src_index" to the list of inputs in the node_def_.
170 NodeDef node_def_; member in class:tensorflow::NodeDefBuilder
|
| shape_inference.cc | 41 node_def_(CHECK_NOTNULL(node_def)) {
94 node_def_(CHECK_NOTNULL(node_def)) {
144 node_def_(CHECK_NOTNULL(node_def)) {
163 << i << " for " << node_def_->name() << " of type " << node_def_->op();
216 string InferenceContext::op() const { return node_def_->op(); }
224 construction_status_ = NameRangesForNode(*node_def_, op_def, &input_name_map_,
326 ProtoDebugString(*node_def_));
[all...] |
| shape_inference.h | 329 AttrSlice attrs() const { return AttrSlice(*node_def_); }
737 const NodeDef* node_def_; member in class:tensorflow::shape_inference::InferenceContext
784 return GetNodeAttr(*node_def_, attr_name, value);
|
| /external/tensorflow/tensorflow/c/eager/ |
| runtime.cc | 143 if (include_those_in_node_def && node_def_ != nullptr) {
144 for (AttrValueMap::const_iterator it = node_def_->attr().begin();
145 it != node_def_->attr().end(); ++it) {
152 if (node_def_finalized_) return *node_def_;
155 node_def_->add_input("dummy_input");
157 FillAttrValueMap(node_def_->mutable_attr(), false);
159 return *node_def_;
191 if (node_def_ != nullptr) {
192 // Some attributes are directly written to node_def_ instead of being
195 for (const auto& attr : node_def_->attr())
[all...] |
| runtime.h | 82 node_def_(nullptr),
91 SetInAttrValueMap(node_def_->mutable_attr(), attr_name, value);
133 std::unique_ptr<NodeDef> node_def_; member in class:tensorflow::AttrBuilder
|
| /external/tensorflow/tensorflow/core/kernels/ |
| ops_testutil.h | 90 void set_node_def(const NodeDef& node_def) { node_def_.CopyFrom(node_def); }
93 NodeDef* node_def() { return &node_def_; }
105 node_def_, graph_def_version, &status);
232 NodeDef node_def_; member in class:tensorflow::OpsTestBase
|
| /external/tensorflow/tensorflow/contrib/graph_editor/ |
| transform.py | 145 node_def_ = deepcopy(op._node_def)
150 node_def_.name = name_
161 op_ = tf_ops.Operation(node_def_, info.graph_, [], output_types_,
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| randomized_tests.cc | 135 NodeDef node_def_; member in class:tensorflow::__anon38715::OpTestBuilder
140 node_def_.set_op(op_name);
172 AddNodeAttr(attr_name, std::forward<T>(value), &node_def_);
191 TF_RETURN_IF_ERROR(op_registry->LookUpOpDef(node_def_.op(), &op_def));
194 *test_def = node_def_;
[all...] |
| /external/tensorflow/tensorflow/core/common_runtime/ |
| shape_refiner.cc | 195 nested_inferences[node_name]->get_context()->node_def_ =
[all...] |
