| /external/tensorflow/tensorflow/core/framework/ |
| D | node_def_builder.cc | 43 node_def_.set_name(string(name)); in NodeDefBuilder()
51 if (debug != nullptr) MergeDebugInfo(*debug, &node_def_); in NodeDefBuilder()
57 MergeDebugInfo(debug, &node_def_); in NodeDefBuilder()
62 node_def_.set_name(string(name)); in NodeDefBuilder()
68 node_def_.set_op(op_def_->name()); in Initialize()
90 Status status = fake_input(*op_def_, inputs_specified_, node_def_, this); in Input()
180 node_def_.add_input(strings::StrCat(src_node, ":", src_index)); in AddInput()
182 node_def_.add_input(string(src_node)); in AddInput()
210 node_def_.set_device(string(device_spec)); in Device()
232 node_def_.name(), "'"); in Finalize()
[all …]
|
| D | fake_input.cc | 48 const NodeDef* const node_def_; member in tensorflow::__anonb827cc640111::FakeInputImpl
64 node_def_(node_def), in FakeInputImpl()
104 Status status = GetNodeAttr(*node_def_, arg_->type_list_attr(), &dts); in AddInputToBuilder()
131 Status status = GetNodeAttr(*node_def_, arg_->number_attr(), n); in GetN()
148 Status status = GetNodeAttr(*node_def_, arg_->type_attr(), dt); in GetDataType()
|
| D | shape_inference.cc | 41 node_def_(CHECK_NOTNULL(node_def)) { in InferenceContext()
96 node_def_(CHECK_NOTNULL(node_def)) { in InferenceContext()
148 node_def_(CHECK_NOTNULL(node_def)) { in InferenceContext()
169 << i << " for " << node_def_->name() << " of type " << node_def_->op(); in Run()
222 string InferenceContext::op() const { return node_def_->op(); } in op()
230 construction_status_ = NameRangesForNode(*node_def_, op_def, &input_name_map_, in PreInputInit()
349 ProtoDebugString(*node_def_)); in DebugString()
1178 " for '", node_def_->name(), "' (op: '", node_def_->op(), in AttachContext()
|
| D | node_def_builder.h | 137 const string& node_name() const { return node_def_.name(); } in node_name()
174 NodeDef node_def_; variable
|
| D | shape_inference.h | 336 AttrSlice attrs() const { return AttrSlice(*node_def_); } in attrs()
773 const NodeDef* node_def_; variable
820 return GetNodeAttr(*node_def_, attr_name, value); in GetAttr()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | attr_builder.cc | 159 if (include_those_in_node_def && node_def_ != nullptr) { in FillAttrValueMap()
160 for (AttrValueMap::const_iterator it = node_def_->attr().begin(); in FillAttrValueMap()
161 it != node_def_->attr().end(); ++it) { in FillAttrValueMap()
184 if (node_def_finalized_) return *node_def_; in BuildNodeDef()
187 node_def_->add_input("dummy_input"); in BuildNodeDef()
189 FillAttrValueMap(node_def_->mutable_attr(), false); in BuildNodeDef()
191 return *node_def_; in BuildNodeDef()
248 if (node_def_ != nullptr) { in BuildCacheKeyForDevice()
252 for (const auto& attr : node_def_->attr()) { in BuildCacheKeyForDevice()
283 if (node_def_ == nullptr) { in MayBeInitializeNodeDef()
[all …]
|
| D | attr_builder.h | 90 node_def_(nullptr), in AttrBuilder()
99 SetInAttrValueMap(node_def_->mutable_attr(), string(attr_name), value); in Set()
143 std::unique_ptr<NodeDef> node_def_; variable
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ops_testutil.h | 90 void set_node_def(const NodeDef& node_def) { node_def_.CopyFrom(node_def); } in set_node_def()
93 NodeDef* node_def() { return &node_def_; } in node_def()
105 node_def_, graph_def_version, &status); in InitOpWithGraphVersion()
231 NodeDef node_def_; variable
|
| /external/tensorflow/tensorflow/contrib/graph_editor/ |
| D | 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/core/kernels/data/ |
| D | concatenate_dataset_op_test.cc | 51 node_def_ = test::function::NDef( in CreateConcatenateDatasetKernel()
54 TF_RETURN_IF_ERROR(CreateOpKernel(node_def_, op_kernel)); in CreateConcatenateDatasetKernel()
69 NodeDef node_def_; member in tensorflow::data::__anon516b62740111::ConcatenateDatasetOpTest
|
| D | zip_dataset_op_test.cc | 61 node_def_ = test::function::NDef( in CreateZipDatasetKernel()
64 TF_RETURN_IF_ERROR(CreateOpKernel(node_def_, op_kernel)); in CreateZipDatasetKernel()
79 NodeDef node_def_; member in tensorflow::data::__anon6eabbe220111::ZipDatasetOpTest
|
| D | tensor_dataset_op_test.cc | 48 node_def_ = test::function::NDef( in CreateTensorDatasetKernel()
51 TF_RETURN_IF_ERROR(CreateOpKernel(node_def_, tensor_dataset_kernel)); in CreateTensorDatasetKernel()
66 NodeDef node_def_; member in tensorflow::data::__anond871d70a0111::TensorDatasetOpTest
|
| D | sparse_tensor_slice_dataset_op_test.cc | 42 node_def_ = test::function::NDef(kNodeName, kOpName, in CreateSparseTensorSliceDatasetKernel()
45 TF_RETURN_IF_ERROR(CreateOpKernel(node_def_, op_kernel)); in CreateSparseTensorSliceDatasetKernel()
59 NodeDef node_def_; member in tensorflow::data::__anon3d973c260111::SparseTensorSliceDatasetOpTest
|
| D | take_dataset_op_test.cc | 41 node_def_ = test::function::NDef( in CreateTakeDatasetKernel()
44 TF_RETURN_IF_ERROR(CreateOpKernel(node_def_, op_kernel)); in CreateTakeDatasetKernel()
58 NodeDef node_def_; member in tensorflow::data::__anon0bec85d40111::TakeDatasetOpTest
|
| D | repeat_dataset_op_test.cc | 44 node_def_ = test::function::NDef( in CreateRepeatDatasetKernel()
47 TF_RETURN_IF_ERROR(CreateOpKernel(node_def_, op_kernel)); in CreateRepeatDatasetKernel()
61 NodeDef node_def_; member in tensorflow::data::__anonf4d903700111::RepeatDatasetOpTest
|
| D | tensor_slice_dataset_op_test.cc | 49 node_def_ = test::function::NDef( in CreateTensorSliceDatasetKernel()
52 TF_RETURN_IF_ERROR(CreateOpKernel(node_def_, tensor_dataset_kernel)); in CreateTensorSliceDatasetKernel()
68 NodeDef node_def_; member in tensorflow::data::__anona47dbc190111::TensorSliceDatasetOpTest
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 144 NodeDef node_def_; member in tensorflow::__anonf5fb02740111::OpTestBuilder
149 node_def_.set_op(op_name); in OpTestBuilder()
193 AddNodeAttr(attr_name, std::forward<T>(value), &node_def_); in Attr()
212 TF_RETURN_IF_ERROR(op_registry->LookUpOpDef(node_def_.op(), &op_def)); in BuildGraph()
215 *test_def = node_def_; in BuildGraph()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner.cc | 196 nested_inferences[node_name]->get_context()->node_def_ = in InferShapesForFunction()
|