| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| resource_operation_table_test.cc | 31 bool HasResourceInputOrOutput(const OpDef& op_def) {
32 return absl::c_any_of(op_def.input_arg(), IsResourceArgDef) ||
33 absl::c_any_of(op_def.output_arg(), IsResourceArgDef);
46 const OpDef* op_def; local
47 TF_ASSERT_OK(OpRegistry::Global()->LookUpOpDef(xla_op_name, &op_def));
48 if (HasResourceInputOrOutput(*op_def)) {
|
| xla_op_registry.cc | 43 const OpDef* op_def; local
44 TF_RETURN_IF_ERROR(OpRegistry::Global()->LookUpOpDef("XlaLaunch", &op_def));
210 const OpDef* op_def; local
211 Status lookup_status = op_registry->LookUpOpDef(op_name, &op_def);
222 for (const OpDef::AttrDef& attr_def : op_def->attr()) {
268 const OpDef::AttrDef& op_def_attr = *FindAttr(type_attr, *op_def);
361 const NodeDef& node_def, const OpKernel* op_kernel, const OpDef* op_def,
365 DCHECK(op_def != nullptr || op_kernel != nullptr);
399 if (op_def) {
402 NameRangesForNode(node_def, *op_def, &input_name_ranges, nullptr))
[all...] |
| /external/tensorflow/tensorflow/core/framework/ |
| 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 | 47 return op_reg_data.op_def;
58 OpDef op_def = op_reg_data.op_def; local
66 InferenceContext c(TF_GRAPH_DEF_VERSION, &def, op_def, {S({}), S({10})}, {},
78 OpDef op_def = op_reg_data.op_def; local
85 InferenceContext c(TF_GRAPH_DEF_VERSION, &def, op_def, {S({})}, {}, {}, {});
92 InferenceContext c(TF_GRAPH_DEF_VERSION, &def, op_def,
109 OpDef op_def = op_reg_data.op_def; local
224 OpDef op_def = op_reg_data.op_def; local
326 OpDef op_def = op_reg_data.op_def; local
[all...] |
| memory_types.cc | 80 const OpDef* op_def; local
81 TF_RETURN_IF_ERROR(op_registry->LookUpOpDef(ndef.op(), &op_def));
91 InOutTypesForNode(ndef, *op_def, &inp_dtypes, &out_dtypes));
111 TF_RETURN_IF_ERROR(NameRangesForNode(ndef, *op_def, &inp_names, &out_names));
125 "' not found in OpDef: ", SummarizeOpDef(*op_def));
|
| node_def_builder.h | 25 #include "tensorflow/core/framework/op_def.pb.h"
71 // REQUIRES: in addition, *op_def must outlive *this.
72 NodeDefBuilder(StringPiece name, const OpDef* op_def);
138 const OpDef& op_def() const { return *op_def_; } function in class:tensorflow::NodeDefBuilder
|
| op_def_builder.h | 24 #include "tensorflow/core/framework/op_def.pb.h"
42 OpRegistrationData(const OpDef& def) : op_def(def) {}
45 : op_def(def), shape_inference_fn(fn), is_function_op(is_function) {}
47 OpDef op_def; member in struct:tensorflow::OpRegistrationData
145 // Sets op_reg_data->op_def to the requested OpDef and
162 OpDef* op_def() { return &op_reg_data_.op_def; } function in class:tensorflow::OpDefBuilder
|
| graph_def_util.cc | 72 const OpDef* op_def; local
73 Status s = op_registry.LookUpOpDef(node_def->op(), &op_def);
75 AddDefaultsToNodeDef(*op_def, node_def);
220 const OpDef* op_def; local
221 TF_RETURN_IF_ERROR(op_registry.LookUpOpDef(op_name, &op_def));
223 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 | 45 const OpDef* op_def; local
46 TF_CHECK_OK(OpRegistry::Global()->LookUpOpDef(node_def()->op(), &op_def));
47 return op_def;
225 TF_ASSERT_OK(NodeDefBuilder("add_attr", &old_op.op_def).Finalize(node_def()));
226 ExpectSuccess(old_op.op_def);
240 TF_ASSERT_OK(NodeDefBuilder("less_strict", &old_op.op_def)
243 ExpectSuccess(old_op.op_def);
258 TF_ASSERT_OK(NodeDefBuilder("remove_restriction", &old_op.op_def)
261 ExpectSuccess(old_op.op_def);
277 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);
|
| function_test.cc | 994 const OpDef* op_def; local
995 EXPECT_TRUE(!lib_def.LookUpOpDef("XTimes16", &op_def).ok());
997 TF_EXPECT_OK(lib_def.LookUpOpDef("XTimesTwo", &op_def));
998 ASSERT_NE(op_def, nullptr);
999 EXPECT_EQ(op_def->DebugString(),
[all...] |
| node_def_util_test.cc | 37 return op_reg_data.op_def;
52 void ExpectSuccess(const NodeDef& good, const OpDef& op_def) {
53 EXPECT_EQ(Status::OK(), ValidateNodeDef(good, op_def))
55 << "; OpDef: " << SummarizeOpDef(op_def);
58 void ExpectFailure(const NodeDef& bad, const OpDef& op_def,
60 Status status = ValidateNodeDef(bad, op_def);
63 << "; OpDef: " << SummarizeOpDef(op_def);
68 << "; OpDef: " << SummarizeOpDef(op_def);
73 << "; OpDef: " << SummarizeOpDef(op_def) << "\nActual error: " << status
374 const OpDef op_def = ToOpDef(OpDefBuilder("Simple" local
395 const OpDef op_def = ToOpDef(OpDefBuilder("Simple") local
416 const OpDef op_def = ToOpDef(OpDefBuilder("Simple") local
436 const OpDef op_def = ToOpDef(OpDefBuilder("Polymorphic") local
462 const OpDef op_def = ToOpDef(OpDefBuilder("NRepeats") local
502 const OpDef op_def = ToOpDef(OpDefBuilder("TypeList") local
[all...] |
| dataset.cc | 280 TF_RETURN_IF_ERROR(AddFunction(ctx, op_reg_data->op_def.name()));
311 const OpDef* op_def = nullptr; local
312 Status s = b_->opts().op_registry()->LookUpOpDef(name, &op_def);
313 if (!s.ok() || op_def == nullptr) {
316 return HasAttr(op_def, attr_name);
|
| /external/tensorflow/tensorflow/core/ops/ |
| tpu_embedding_ops.cc | 88 auto* op_def = &op_reg_data->op_def; local
89 op_def->set_name(
97 auto* arg = op_def->add_input_arg();
103 auto* table_id_attr = op_def->add_attr();
111 auto* table_name_attr = op_def->add_attr();
117 auto* num_shards_attr = op_def->add_attr();
122 auto* shard_id_attr = op_def->add_attr();
137 op_def->set_is_commutative(false);
138 op_def->set_is_aggregate(false)
205 auto* op_def = &op_reg_data->op_def; local
[all...] |
| /external/tensorflow/tensorflow/js/ops/ |
| ts_op_gen_test.cc | 18 #include "tensorflow/core/framework/op_def.pb.h"
146 const string op_def = R"( local
176 GenerateTsOpFileText(op_def, "", &ts_file_text);
182 const string op_def = R"( local
216 GenerateTsOpFileText(op_def, "", &ts_file_text);
|
| /external/tensorflow/tensorflow/core/graph/ |
| node_builder.h | 22 #include "tensorflow/core/framework/op_def.pb.h"
82 NodeBuilder(StringPiece name, const OpDef* op_def);
128 const OpDef& op_def() const { return def_builder_.op_def(); } function in class:tensorflow::NodeBuilder
|
| validate.cc | 37 const OpDef* op_def; local
38 TF_RETURN_IF_ERROR(op_registry.LookUpOpDef(node_def.op(), &op_def));
39 TF_RETURN_IF_ERROR(ValidateNodeDef(node_def, *op_def));
40 TF_RETURN_IF_ERROR(CheckOpDeprecation(*op_def, version));
|
| /external/tensorflow/tensorflow/c/eager/ |
| c_api_internal.h | 103 explicit TFE_OpInferenceContext(const tensorflow::OpDef* op_def)
104 : op_def(op_def) {}
106 const tensorflow::OpDef* op_def; // op definition from protobuf member in struct:TFE_OpInferenceContext
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| attr_builder.cc | 56 Status OpDefForOp(const char* op_name, const OpDef** op_def) {
60 *op_def = &op_reg_data->op_def;
71 const OpDef* op_def = nullptr; local
72 Status s = OpDefForOp(op_name, &op_def);
89 for (const auto& attr : op_def->attr()) {
169 const OpDef* op_def = nullptr; local
170 Status s = OpDefForOp(op_name_.c_str(), &op_def);
175 DCHECK(op_def);
176 for (const auto& attr_def : op_def->attr())
[all...] |
| kernel_and_device.cc | 82 const OpDef* op_def = nullptr; local
96 op_def = &(function_def->signature());
98 TF_RETURN_IF_ERROR(OpDefForOp(ndef.op().c_str(), &op_def));
101 InOutTypesForNode(ndef, *op_def, &input_dtypes_, &output_dtypes_));
|
| execute.cc | 227 const OpDef* op_def = nullptr; local
232 op_def = &(function_def->signature());
234 TF_RETURN_IF_ERROR(OpDefForOp(op->Name().c_str(), &op_def));
237 TF_RETURN_IF_ERROR(OutputTypesForNode(node_def, *op_def, output_dtypes));
[all...] |
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| eager_service_impl.cc | 56 const tensorflow::OpDef& op_def = op_reg_data->op_def; local
58 for (const auto& output_arg : op_def.output_arg()) {
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| hoist_random_uniform.cc | 21 #include "tensorflow/core/framework/op_def.pb.h"
190 const OpDef* op_def; local
191 TF_CHECK_OK(library.LookUpOpDef(node.op(), &op_def));
192 if (!op_def->is_stateful()) continue;
202 if (op_def->name() != "RandomUniform") return false;
|
| /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;
|
