| /external/tensorflow/tensorflow/core/framework/ |
| node_def_builder.cc | 43 node_def_.set_name(string(name));
51 if (debug != nullptr) MergeDebugInfo(*debug, &node_def_);
57 MergeDebugInfo(debug, &node_def_);
62 node_def_.set_name(string(name));
68 node_def_.set_op(op_def_->name());
90 Status status = fake_input(*op_def_, inputs_specified_, node_def_, this);
180 node_def_.add_input(strings::StrCat(src_node, ":", src_index));
182 node_def_.add_input(string(src_node));
210 node_def_.set_device(string(device_spec));
232 node_def_.name(), "'")
[all...] |
| fake_input.cc | 48 const NodeDef* const node_def_; member in class:tensorflow::__anon44960::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 | 137 const string& node_name() const { return node_def_.name(); }
158 // Add "src_node:src_index" to the list of inputs in the node_def_.
174 NodeDef node_def_; member in class:tensorflow::NodeDefBuilder
|
| shape_inference.cc | 41 node_def_(CHECK_NOTNULL(node_def)) {
96 node_def_(CHECK_NOTNULL(node_def)) {
148 node_def_(CHECK_NOTNULL(node_def)) {
169 << i << " for " << node_def_->name() << " of type " << node_def_->op();
222 string InferenceContext::op() const { return node_def_->op(); }
230 construction_status_ = NameRangesForNode(*node_def_, op_def, &input_name_map_,
349 ProtoDebugString(*node_def_));
[all...] |
| shape_inference.h | 336 AttrSlice attrs() const { return AttrSlice(*node_def_); }
773 const NodeDef* node_def_; member in class:tensorflow::shape_inference::InferenceContext
[all...] |
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| attr_builder.cc | 159 if (include_those_in_node_def && node_def_ != nullptr) {
160 for (AttrValueMap::const_iterator it = node_def_->attr().begin();
161 it != node_def_->attr().end(); ++it) {
184 if (node_def_finalized_) return *node_def_;
187 node_def_->add_input("dummy_input");
189 FillAttrValueMap(node_def_->mutable_attr(), false);
191 return *node_def_;
248 if (node_def_ != nullptr) {
249 // Some attributes are directly written to node_def_ instead of being
252 for (const auto& attr : node_def_->attr())
[all...] |
| attr_builder.h | 90 node_def_(nullptr),
99 SetInAttrValueMap(node_def_->mutable_attr(), string(attr_name), value);
121 // from `node_def_`.
143 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);
231 NodeDef node_def_; member in class:tensorflow::OpsTestBase
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| concatenate_dataset_op_test.cc | 51 node_def_ = test::function::NDef(
54 TF_RETURN_IF_ERROR(CreateOpKernel(node_def_, op_kernel));
69 NodeDef node_def_; member in class:tensorflow::data::__anon45261::ConcatenateDatasetOpTest
|
| zip_dataset_op_test.cc | 61 node_def_ = test::function::NDef(
64 TF_RETURN_IF_ERROR(CreateOpKernel(node_def_, op_kernel));
79 NodeDef node_def_; member in class:tensorflow::data::__anon45345::ZipDatasetOpTest
|
| repeat_dataset_op_test.cc | 44 node_def_ = test::function::NDef(
47 TF_RETURN_IF_ERROR(CreateOpKernel(node_def_, op_kernel));
61 NodeDef node_def_; member in class:tensorflow::data::__anon45327::RepeatDatasetOpTest
|
| sparse_tensor_slice_dataset_op_test.cc | 42 node_def_ = test::function::NDef(kNodeName, kOpName,
45 TF_RETURN_IF_ERROR(CreateOpKernel(node_def_, op_kernel));
59 NodeDef node_def_; member in class:tensorflow::data::__anon45334::SparseTensorSliceDatasetOpTest
|
| tensor_dataset_op_test.cc | 48 node_def_ = test::function::NDef(
51 TF_RETURN_IF_ERROR(CreateOpKernel(node_def_, tensor_dataset_kernel));
66 NodeDef node_def_; member in class:tensorflow::data::__anon45338::TensorDatasetOpTest
|
| take_dataset_op_test.cc | 41 node_def_ = test::function::NDef(
44 TF_RETURN_IF_ERROR(CreateOpKernel(node_def_, op_kernel));
58 NodeDef node_def_; member in class:tensorflow::data::__anon45336::TakeDatasetOpTest
|
| tensor_slice_dataset_op_test.cc | 49 node_def_ = test::function::NDef(
52 TF_RETURN_IF_ERROR(CreateOpKernel(node_def_, tensor_dataset_kernel));
68 NodeDef node_def_; member in class:tensorflow::data::__anon45340::TensorSliceDatasetOpTest
|
| /external/tensorflow/tensorflow/contrib/graph_editor/ |
| transform.py | 155 node_def_ = deepcopy(op.node_def)
160 node_def_.name = name_
164 node_def_ = nodedef_fn(node_def_)
175 op_ = tf_ops.Operation(node_def_, info.graph_, new_inputs, output_types_,
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| randomized_tests.cc | 144 NodeDef node_def_; member in class:tensorflow::__anon44094::OpTestBuilder
149 node_def_.set_op(op_name);
193 AddNodeAttr(attr_name, std::forward<T>(value), &node_def_);
212 TF_RETURN_IF_ERROR(op_registry->LookUpOpDef(node_def_.op(), &op_def));
215 *test_def = node_def_;
[all...] |
| /external/tensorflow/tensorflow/core/common_runtime/ |
| shape_refiner.cc | 196 nested_inferences[node_name]->get_context()->node_def_ =
|
