HomeSort by relevance Sort by last modified time
    Searched refs:op_def (Results 1 - 25 of 134) sorted by null

1 2 3 4 5 6

  /external/tensorflow/tensorflow/python/framework/
op_def_registry.py 33 for op_def in op_list.op:
34 if op_def.name in _registered_ops:
35 assert _registered_ops[op_def.name] == op_def
37 _registered_ops[op_def.name] = op_def
op_def_library.py 37 def _Attr(op_def, name):
38 for attr in op_def.attr:
42 (op_def.name, name))
234 def __init__(self, op_def):
235 self.op_def = op_def
238 for arg in list(op_def.input_arg) + list(op_def.output_arg):
242 (arg.name, op_def.name, num_type_fields))
244 attr_type = _Attr(op_def, arg.type_attr).typ
    [all...]
c_api_util_test.py 34 op_def = api_def_map.get_op_def("Add")
35 self.assertEqual(op_def.name, "Add")
  /external/tensorflow/tensorflow/core/framework/
op_def_util.h 24 #include "tensorflow/core/framework/op_def.pb.h"
30 // Performs a consistency check across the fields of the op_def.
31 Status ValidateOpDef(const OpDef& op_def);
35 Status CheckOpDeprecation(const OpDef& op_def, int graph_def_version);
42 // The following search through op_def for an attr with the indicated name.
44 const OpDef::AttrDef* FindAttr(StringPiece name, const OpDef& op_def);
45 OpDef::AttrDef* FindAttrMutable(StringPiece name, OpDef* op_def);
47 // Searches op_def for input argument with the indicated name.
49 const OpDef::ArgDef* FindInputArg(StringPiece name, const OpDef& op_def);
55 // Produce a human-readable version of an op_def that is more concis
    [all...]
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_builder.cc 108 strings::StrCat(__VA_ARGS__, AttrError(orig, op_def->name()))); \
146 void FinalizeAttr(StringPiece spec, OpDef* op_def,
148 OpDef::AttrDef* attr = op_def->add_attr();
331 __VA_ARGS__, InOutError(is_output, orig, op_def->name()))); \
336 void FinalizeInputOrOutput(StringPiece spec, bool is_output, OpDef* op_def,
339 is_output ? op_def->add_output_arg() : op_def->add_input_arg();
367 const OpDef::AttrDef* attr = FindAttr(type_or_attr, *op_def);
392 OpDef::AttrDef* attr = FindAttrMutable(arg->number_attr(), op_def);
400 OpDef::AttrDef* attr = FindAttrMutable(arg->type_list_attr(), op_def);
639 OpDef* op_def = &op_reg_data->op_def; local
    [all...]
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
op_def_util.cc 23 #include "tensorflow/core/framework/op_def.pb_text.h"
150 const OpDef::AttrDef* FindAttr(StringPiece name, const OpDef& op_def) {
151 for (int i = 0; i < op_def.attr_size(); ++i) {
152 if (op_def.attr(i).name() == name) {
153 return &op_def.attr(i);
159 OpDef::AttrDef* FindAttrMutable(StringPiece name, OpDef* op_def) {
160 for (int i = 0; i < op_def->attr_size(); ++i) {
161 if (op_def->attr(i).name() == name) {
162 return op_def->mutable_attr(i);
168 const OpDef::ArgDef* FindInputArg(StringPiece name, const OpDef& op_def) {
763 OpDef* op_def = op_list->mutable_op(i); local
    [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...]
fake_input.cc 21 #include "tensorflow/core/framework/op_def.pb.h"
31 FakeInputImpl(const OpDef* op_def, int in_index, const NodeDef* node_def,
59 FakeInputImpl::FakeInputImpl(const OpDef* op_def, int in_index,
61 : op_def_(op_def),
62 arg_(&op_def->input_arg(in_index)),
193 return [](const OpDef& op_def, int in_index, const NodeDef& node_def,
195 FakeInputImpl impl(&op_def, in_index, &node_def, builder);
201 return [dt](const OpDef& op_def, int in_index, const NodeDef& node_def,
203 FakeInputImpl impl(&op_def, in_index, &node_def, builder);
210 return [n](const OpDef& op_def, int in_index, const NodeDef& node_def
    [all...]
op.cc 38 const OpDef** op_def) const {
39 *op_def = nullptr;
42 *op_def = &op_reg_data->op_def;
129 op_defs->push_back(p.second->op_def);
166 *out->Add() = item.second->op_def;
224 s = ValidateOpDef(op_reg_data->op_def);
226 !gtl::InsertIfNotPresent(&registry_, op_reg_data->op_def.name(),
228 s = errors::AlreadyExists("Op with name ", op_reg_data->op_def.name());
233 watcher_status = watcher_(s, 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...]
node_def_util.cc 26 #include "tensorflow/core/framework/op_def.pb_text.h"
402 Status InputTypeForNode(const NodeDef& node_def, const OpDef& op_def,
405 for (const auto& arg : op_def.input_arg()) {
417 Status InputTypesForNode(const NodeDef& node_def, const OpDef& op_def,
419 for (const auto& arg : op_def.input_arg()) {
425 Status OutputTypeForNode(const NodeDef& node_def, const OpDef& op_def,
428 for (const auto& arg : op_def.output_arg()) {
440 Status OutputTypesForNode(const NodeDef& node_def, const OpDef& op_def,
442 for (const auto& arg : op_def.output_arg()) {
448 Status InOutTypesForNode(const NodeDef& node_def, const OpDef& op_def,
    [all...]
op_registration_test.cc 29 op_reg_data->op_def.set_name(op_name);
52 registry->SetWatcher([](const Status& s, const OpDef& op_def) -> Status {
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));
  /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)) {
  /external/tensorflow/tensorflow/core/ops/
shape_function_test.cc 30 << op_reg_data.op_def.name();
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.cc 53 GenTypeScriptOp(const OpDef& op_def, const ApiDef& api_def);
89 GenTypeScriptOp::GenTypeScriptOp(const OpDef& op_def, const ApiDef& api_def)
90 : op_def_(op_def), api_def_(api_def), num_outputs_(0) {}
229 void WriteTSOp(const OpDef& op_def, const ApiDef& api_def, WritableFile* ts) {
230 GenTypeScriptOp ts_op(op_def, api_def);
231 TF_CHECK_OK(ts->Append(GenTypeScriptOp(op_def, api_def).Code()));
274 for (const auto& op_def : ops.op()) {
276 if (op_def.has_deprecation() &&
277 op_def.deprecation().version() <= TF_GRAPH_DEF_VERSION) {
281 const auto* api_def = api_def_map.GetApiDef(op_def.name())
    [all...]
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/
validate_test.cc 102 *op_list.add_op() = op_reg_data.op_def;
114 *op_list.add_op() = op_reg_data.op_def;
130 for (const OpDef& op_def : op_list.op()) {
131 if (op_def.name() == "FloatInput") {
135 if (op_def.name() == "Int32Input") {
139 if (op_def.name() == "HasDocs") {
142 EXPECT_TRUE(op_def.summary().empty());
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/java/src/gen/cc/
op_specs.cc 46 explicit TypeResolver(const OpDef& op_def) : op_def_(op_def) {}
335 EndpointSpec CreateEndpoint(const OpDef& op_def, const ApiDef& api_def,
354 OpSpec OpSpec::Create(const OpDef& op_def, const ApiDef& api_def) {
356 op_def.deprecation().explanation());
357 TypeResolver type_resolver(op_def);
359 for (int i = 0; i < op_def.input_arg().size(); ++i) {
360 if (op_def.input_arg(i).name() == next_input_name) {
361 op.inputs_.push_back(CreateInput(op_def.input_arg(i), api_def.in_arg(i),
367 for (int i = 0; i < op_def.attr().size(); ++i)
    [all...]
  /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...]

Completed in 1056 milliseconds

1 2 3 4 5 6