/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(®istry_, 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...] |