| /external/tensorflow/tensorflow/core/graph/ |
| validate.cc | 35 const OpDef* op_def; local
36 TF_RETURN_IF_ERROR(op_registry.LookUpOpDef(node_def.op(), &op_def));
37 TF_RETURN_IF_ERROR(ValidateNodeDef(node_def, *op_def));
38 TF_RETURN_IF_ERROR(CheckOpDeprecation(*op_def, version));
|
| node_builder.h | 22 #include "tensorflow/core/framework/op_def.pb.h"
80 NodeBuilder(StringPiece name, const OpDef* op_def);
120 const OpDef& op_def() const { return def_builder_.op_def(); } function in class:tensorflow::NodeBuilder
|
| graph_constructor_test.cc | 2624 const OpDef* op_def; local
2719 const OpDef* op_def; local
[all...] |
| /external/tensorflow/tensorflow/core/framework/ |
| op_def_builder.h | 24 #include "tensorflow/core/framework/op_def.pb.h"
40 OpRegistrationData(const OpDef& def) : op_def(def) {}
43 : op_def(def), shape_inference_fn(fn), is_function_op(is_function) {}
45 OpDef op_def; member in struct:tensorflow::OpRegistrationData
143 // Sets op_reg_data->op_def to the requested OpDef and
153 OpDef* op_def() { return &op_reg_data_.op_def; } function in class:tensorflow::OpDefBuilder
|
| op_def_builder_test.cc | 19 #include "tensorflow/core/framework/op_def.pb.h"
48 OpDef& op_def = op_reg_data.op_def; local
54 CanonicalizeAttrTypeListOrder(&op_def);
56 EXPECT_EQ(op_def.ShortDebugString(), expected.ShortDebugString());
68 OpDef& op_def = op_reg_data.op_def; local
73 EXPECT_EQ(op_def.ShortDebugString(), expected.ShortDebugString());
632 const OpDef& op_def = op_reg_data.op_def;
[all...] |
| common_shape_fns_test.cc | 46 return op_reg_data.op_def;
57 OpDef op_def = op_reg_data.op_def; local
65 InferenceContext c(TF_GRAPH_DEF_VERSION, &def, op_def, {S({}), S({10})}, {},
77 OpDef op_def = op_reg_data.op_def; local
84 InferenceContext c(TF_GRAPH_DEF_VERSION, &def, op_def, {S({})}, {}, {}, {});
91 InferenceContext c(TF_GRAPH_DEF_VERSION, &def, op_def,
108 OpDef op_def = op_reg_data.op_def; local
226 OpDef op_def = op_reg_data.op_def; local
330 OpDef op_def = op_reg_data.op_def; local
[all...] |
| memory_types.cc | 75 const OpDef* op_def; local
76 TF_RETURN_IF_ERROR(op_registry->LookUpOpDef(ndef.op(), &op_def));
86 InOutTypesForNode(ndef, *op_def, &inp_dtypes, &out_dtypes));
106 TF_RETURN_IF_ERROR(NameRangesForNode(ndef, *op_def, &inp_names, &out_names));
120 "' not found in OpDef: ", SummarizeOpDef(*op_def));
|
| node_def_builder.h | 25 #include "tensorflow/core/framework/op_def.pb.h"
67 // REQUIRES: in addition, *op_def must outlive *this.
68 NodeDefBuilder(StringPiece name, const OpDef* op_def);
134 const OpDef& op_def() const { return *op_def_; } function in class:tensorflow::NodeDefBuilder
|
| graph_def_util.cc | 65 const OpDef* op_def; local
66 TF_RETURN_IF_ERROR(op_registry.LookUpOpDef(node_def->op(), &op_def));
67 AddDefaultsToNodeDef(*op_def, node_def);
209 const OpDef* op_def; local
210 TF_RETURN_IF_ERROR(op_registry.LookUpOpDef(op_name, &op_def));
212 stripped_op->CopyFrom(*op_def);
|
| graph_def_util_test.cc | 22 #include "tensorflow/core/framework/op_def.pb.h"
31 Status FinalizeOpDef(const OpDefBuilder& b, OpDef* op_def) {
34 *op_def = op_reg_data.op_def;
231 OpDef* op_def = op_list.add_op(); local
232 op_def->set_name(op);
233 op_def->set_summary("summary");
234 op_def->set_description("description");
235 op_def->set_is_commutative(op == "B");
|
| op_compatibility_test.cc | 44 const OpDef* op_def; local
45 TF_CHECK_OK(OpRegistry::Global()->LookUpOpDef(node_def()->op(), &op_def));
46 return op_def;
224 TF_ASSERT_OK(NodeDefBuilder("add_attr", &old_op.op_def).Finalize(node_def()));
225 ExpectSuccess(old_op.op_def);
239 TF_ASSERT_OK(NodeDefBuilder("less_strict", &old_op.op_def)
242 ExpectSuccess(old_op.op_def);
257 TF_ASSERT_OK(NodeDefBuilder("remove_restriction", &old_op.op_def)
260 ExpectSuccess(old_op.op_def);
275 TF_ASSERT_OK(NodeDefBuilder("attr_order", &old_op.op_def)
[all...] |
| op_def_util_test.cc | 18 #include "tensorflow/core/framework/op_def.pb.h"
30 OpDef op_def; local
31 EXPECT_TRUE(protobuf::TextFormat::MergeFromString(text, &op_def));
32 return op_def;
52 return ValidateOpDef(op_reg_data.op_def);
|
| node_def_util_test.cc | 34 return op_reg_data.op_def;
49 void ExpectSuccess(const NodeDef& good, const OpDef& op_def) {
50 EXPECT_EQ(Status::OK(), ValidateNodeDef(good, op_def))
52 << "; OpDef: " << SummarizeOpDef(op_def);
55 void ExpectFailure(const NodeDef& bad, const OpDef& op_def,
57 Status status = ValidateNodeDef(bad, op_def);
60 << "; OpDef: " << SummarizeOpDef(op_def);
65 << "; OpDef: " << SummarizeOpDef(op_def);
70 << "; OpDef: " << SummarizeOpDef(op_def) << "\nActual error: " << status
369 const OpDef op_def = ToOpDef(OpDefBuilder("Simple" local
389 const OpDef op_def = ToOpDef(OpDefBuilder("Polymorphic") local
414 const OpDef op_def = ToOpDef(OpDefBuilder("NRepeats") local
453 const OpDef op_def = ToOpDef(OpDefBuilder("TypeList") local
[all...] |
| dataset.cc | 166 TF_RETURN_IF_ERROR(AddFunction(ctx, op_reg_data->op_def.name()));
190 const OpDef* op_def = nullptr; local
191 Status s = b_->opts().op_registry()->LookUpOpDef(op_type_name, &op_def);
192 if (!s.ok() || op_def == nullptr) {
195 return HasAttr(op_def, attr_name);
|
| function_test.cc | 934 const OpDef* op_def; local
935 EXPECT_TRUE(!lib_def.LookUpOpDef("XTimes16", &op_def).ok());
937 TF_EXPECT_OK(lib_def.LookUpOpDef("XTimesTwo", &op_def));
938 ASSERT_NE(op_def, nullptr);
939 EXPECT_EQ(op_def->DebugString(),
[all...] |
| dataset.h | 179 const OpDef* op_def; local
180 TF_RETURN_IF_ERROR(lib_def->LookUpOpDef(node_def.op(), &op_def));
185 if (IsOpWhitelisted(op_def)) {
188 if (op_def->is_stateful()) {
201 // Also looks up the `op_def->name` in the global
203 bool IsOpWhitelisted(const OpDef* op_def) const {
204 return (StringPiece(op_def->name()).ends_with("Dataset") &&
205 op_def->output_arg_size() == 1 &&
206 op_def->output_arg(0).type() == DT_VARIANT) ||
208 op_def->name())
[all...] |
| op_def_builder.cc | 108 strings::StrCat(__VA_ARGS__, AttrError(orig, op_def->name()))); \
144 void FinalizeAttr(StringPiece spec, OpDef* op_def,
146 OpDef::AttrDef* attr = op_def->add_attr();
318 __VA_ARGS__, InOutError(is_output, orig, op_def->name()))); \
323 void FinalizeInputOrOutput(StringPiece spec, bool is_output, OpDef* op_def,
326 is_output ? op_def->add_output_arg() : op_def->add_input_arg();
354 const OpDef::AttrDef* attr = FindAttr(type_or_attr, *op_def);
379 OpDef::AttrDef* attr = FindAttrMutable(arg->number_attr(), op_def);
387 OpDef::AttrDef* attr = FindAttrMutable(arg->type_list_attr(), op_def);
602 OpDef* op_def = &op_reg_data->op_def; local
[all...] |
| op_def_util.cc | 23 #include "tensorflow/core/framework/op_def.pb_text.h"
146 const OpDef::AttrDef* FindAttr(StringPiece name, const OpDef& op_def) {
147 for (int i = 0; i < op_def.attr_size(); ++i) {
148 if (op_def.attr(i).name() == name) {
149 return &op_def.attr(i);
155 OpDef::AttrDef* FindAttrMutable(StringPiece name, OpDef* op_def) {
156 for (int i = 0; i < op_def->attr_size(); ++i) {
157 if (op_def->attr(i).name() == name) {
158 return op_def->mutable_attr(i);
164 const OpDef::ArgDef* FindInputArg(StringPiece name, const OpDef& op_def) {
749 OpDef* op_def = op_list->mutable_op(i); local
[all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| xla_op_registry.cc | 41 const OpDef* op_def; local
42 TF_RETURN_IF_ERROR(OpRegistry::Global()->LookUpOpDef("_XlaLaunch", &op_def));
163 const OpDef* op_def; local
164 Status lookup_status = op_registry->LookUpOpDef(op_name, &op_def);
174 for (const OpDef::AttrDef& attr_def : op_def->attr()) {
211 const OpDef::AttrDef& op_def_attr = *FindAttr(type_attr, *op_def);
|
| /external/tensorflow/tensorflow/core/grappler/ |
| utils.cc | 22 #include "tensorflow/core/framework/op_def.pb.h"
213 const OpDef* op_def = nullptr; local
214 auto status = OpRegistry::Global()->LookUpOpDef(node.op(), &op_def);
216 for (const auto& output : op_def->output_arg()) {
228 auto status = fdef.LookUpOpDef(node.op(), &op_def);
230 num_outputs = op_def->output_arg_size();
|
| /external/toybox/toys/pending/ |
| expr.c | 135 static struct op_def { struct
154 void eval_op(struct op_def *o, struct value *ret, struct value *rhs)
231 struct op_def *o = OPS;
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| utils.cc | 32 #include "tensorflow/core/framework/op_def.pb.h"
87 const OpDef* op_def = nullptr; local
88 auto s = op_registry->LookUpOpDef(node.op(), &op_def);
90 op_def = nullptr;
127 if (op_def && i < op_def->input_arg_size() &&
128 op_def->input_arg(i).name().find("filename") != std::string::npos) {
153 if (op_def && i < op_def->input_arg_size() &&
154 op_def->input_arg(i).name().find("handle") != std::string::npos)
[all...] |
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| dependency_optimizer.cc | 133 const OpDef* op_def = nullptr; local
134 Status status = OpRegistry::Global()->LookUpOpDef(node.op(), &op_def);
135 if (!status.ok() || op_def->output_arg_size() == 0) {
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| quantize_nodes.cc | 228 const OpDef* op_def = nullptr; local
230 OpRegistry::Global()->LookUpOpDef(current_node->op(), &op_def));
231 const bool is_duplicate = ((!op_def->is_stateful()) && (i > 0));
[all...] |
| /external/tensorflow/tensorflow/c/ |
| c_api_function_test.cc | 20 #include "tensorflow/core/framework/op_def.pb.h"
1495 OpDef op_def; local
1539 OpDef op_def; local
[all...] |
