/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_ =
|