Home
last modified time | relevance | path

Searched refs:OpDef (Results 1 – 25 of 145) sorted by relevance

123456

/external/tensorflow/tensorflow/core/framework/
Dop_def_util.h31 Status ValidateOpDef(const OpDef& op_def);
35 Status CheckOpDeprecation(const OpDef& op_def, int graph_def_version);
40 const OpDef::AttrDef& attr);
44 const OpDef::AttrDef* FindAttr(StringPiece name, const OpDef& op_def);
45 OpDef::AttrDef* FindAttrMutable(StringPiece name, OpDef* op_def);
49 const OpDef::ArgDef* FindInputArg(StringPiece name, const OpDef& op_def);
57 string SummarizeOpDef(const OpDef& op_def);
62 Status OpDefCompatible(const OpDef& old_op, const OpDef& new_op);
67 Status OpDefAddedDefaultsUnchanged(const OpDef& old_op,
68 const OpDef& penultimate_op,
[all …]
Dop_def_util.cc40 bool HasAttrStyleType(const OpDef::ArgDef& arg) { in HasAttrStyleType()
45 Status AllowedTypeValue(DataType dt, const OpDef::AttrDef& attr) { in AllowedTypeValue()
65 Status AllowedStringValue(const string& str, const OpDef::AttrDef& attr) { in AllowedStringValue()
88 const OpDef::AttrDef& attr) { in ValidateAttrValue()
150 const OpDef::AttrDef* FindAttr(StringPiece name, const OpDef& op_def) { in FindAttr()
159 OpDef::AttrDef* FindAttrMutable(StringPiece name, OpDef* op_def) { in FindAttrMutable()
168 const OpDef::ArgDef* FindInputArg(StringPiece name, const OpDef& op_def) { in FindInputArg()
194 static Status ValidateArg(const OpDef::ArgDef& arg, const OpDef& op_def, in ValidateArg()
203 const OpDef::AttrDef* attr = FindAttr(arg.number_attr(), op_def); in ValidateArg()
228 const OpDef::AttrDef* attr = FindAttr(arg.type_attr(), op_def); in ValidateArg()
[all …]
Dop_def_util_test.cc29 OpDef FromText(const string& text) { in FromText()
30 OpDef op_def; in FromText()
35 OpDef::AttrDef ADef(const string& text) { in ADef()
36 OpDef::AttrDef attr_def; in ADef()
356 void ExpectDifferent(const OpDef::AttrDef& a1, const OpDef::AttrDef& a2) { in ExpectDifferent()
363 OpDef::AttrDef a = ADef( in TEST()
428 protobuf::RepeatedPtrField<OpDef::AttrDef> Rep( in Rep()
429 const std::vector<OpDef::AttrDef>& defs) { in Rep()
430 protobuf::RepeatedPtrField<OpDef::AttrDef> rep; in Rep()
431 for (const OpDef::AttrDef& def : defs) { in Rep()
[all …]
Dnode_def_builder.h35 typedef std::function<Status(const OpDef&, int, const NodeDef&,
72 NodeDefBuilder(StringPiece name, const OpDef* op_def);
138 const OpDef& op_def() const { return *op_def_; } in op_def()
146 const OpDef::ArgDef* NextArgDef();
153 void SingleInput(const OpDef::ArgDef* input_arg, StringPiece src_node,
155 void ListInput(const OpDef::ArgDef* input_arg,
162 void VerifyInputType(const OpDef::ArgDef* input_arg, DataType expected,
166 void VerifyInputRef(const OpDef::ArgDef* input_arg, DataType dt);
169 DataType MaybeAddRef(const OpDef::ArgDef* input_arg, DataType dt) { in MaybeAddRef()
173 const OpDef* op_def_;
Dfake_input.cc31 FakeInputImpl(const OpDef* op_def, int in_index, const NodeDef* node_def,
45 const OpDef* const op_def_;
46 const OpDef::ArgDef* const arg_;
59 FakeInputImpl::FakeInputImpl(const OpDef* op_def, int in_index, in FakeInputImpl()
151 const OpDef::AttrDef* attr = FindAttr(arg_->type_attr(), *op_def_); in GetDataType()
193 return [](const OpDef& op_def, int in_index, const NodeDef& node_def, in FakeInput()
201 return [dt](const OpDef& op_def, int in_index, const NodeDef& node_def, in FakeInput()
210 return [n](const OpDef& op_def, int in_index, const NodeDef& node_def, in FakeInput()
219 return [n, dt](const OpDef& op_def, int in_index, const NodeDef& node_def, in FakeInput()
232 return [dtv](const OpDef& op_def, int in_index, const NodeDef& node_def, in FakeInput()
Dnode_def_util_test.cc34 OpDef ToOpDef(const OpDefBuilder& builder) { in ToOpDef()
52 void ExpectSuccess(const NodeDef& good, const OpDef& op_def) { in ExpectSuccess()
58 void ExpectFailure(const NodeDef& bad, const OpDef& op_def, in ExpectFailure()
78 const OpDef op = ToOpDef(OpDefBuilder("In").Input("i: T").Attr("T: type")); in TEST()
142 const OpDef op = in TEST()
163 const OpDef op = ToOpDef(OpDefBuilder("Enum").Attr("e: {'apple','orange'}")); in TEST()
186 const OpDef op = ToOpDef(OpDefBuilder("SameIn") in TEST()
217 const OpDef op = in TEST()
245 const OpDef op_def1 = ToOpDef(OpDefBuilder("None")); in TEST()
252 const OpDef op_def2 = ToOpDef(OpDefBuilder("WithAttr").Attr("v: int")); in TEST()
[all …]
Dnode_def_util.h36 class OpDef; variable
257 Status InputTypeForNode(const NodeDef& node_def, const OpDef& op_def,
261 Status InputTypesForNode(const NodeDef& node_def, const OpDef& op_def,
265 Status OutputTypeForNode(const NodeDef& node_def, const OpDef& op_def,
269 Status OutputTypesForNode(const NodeDef& node_def, const OpDef& op_def,
273 Status InOutTypesForNode(const NodeDef& node_def, const OpDef& op_def,
277 Status NumOutputsForNode(const NodeDef& node_def, const OpDef& op_def,
285 Status ValidateNodeDef(const NodeDef& node_def, const OpDef& op_def);
297 Status NameRangesForNode(const NodeDef& node_def, const OpDef& op_def,
299 Status NameRangesForNode(const Node& node, const OpDef& op_def,
[all …]
Dop_def_builder.h42 OpRegistrationData(const OpDef& def) : op_def(def) {} in OpRegistrationData()
43 OpRegistrationData(const OpDef& def, const OpShapeInferenceFn& fn,
47 OpDef op_def;
162 OpDef* op_def() { return &op_reg_data_.op_def; } in op_def()
Dgraph_def_util.cc72 const OpDef* op_def; in AddDefaultAttrsToGraphDef()
88 const OpDef* producer_op_def; in RemoveNewDefaultAttrsFromNodeDef()
89 const OpDef* consumer_op_def; in RemoveNewDefaultAttrsFromNodeDef()
100 const OpDef::AttrDef* producer_attr_def = in RemoveNewDefaultAttrsFromNodeDef()
220 const OpDef* op_def; in StrippedOpListForGraph()
222 OpDef* stripped_op = stripped_op_list->add_op(); in StrippedOpListForGraph()
Dnode_def_builder.cc60 NodeDefBuilder::NodeDefBuilder(StringPiece name, const OpDef* op_def) in NodeDefBuilder()
71 const OpDef::ArgDef* NodeDefBuilder::NextArgDef() { in NextArgDef()
98 const OpDef::ArgDef* arg = NextArgDef(); in Input()
110 const OpDef::ArgDef* arg = NextArgDef(); in Input()
115 void NodeDefBuilder::SingleInput(const OpDef::ArgDef* input_arg, in SingleInput()
136 void NodeDefBuilder::ListInput(const OpDef::ArgDef* input_arg, in ListInput()
186 void NodeDefBuilder::VerifyInputType(const OpDef::ArgDef* input_arg, in VerifyInputType()
195 void NodeDefBuilder::VerifyInputRef(const OpDef::ArgDef* input_arg, in VerifyInputRef()
Dnode_def_util.cc347 Status AddArgToSig(const NodeDef& node_def, const OpDef::ArgDef& arg_def, in AddArgToSig()
402 Status InputTypeForNode(const NodeDef& node_def, const OpDef& op_def, in InputTypeForNode()
417 Status InputTypesForNode(const NodeDef& node_def, const OpDef& op_def, in InputTypesForNode()
425 Status OutputTypeForNode(const NodeDef& node_def, const OpDef& op_def, in OutputTypeForNode()
440 Status OutputTypesForNode(const NodeDef& node_def, const OpDef& op_def, in OutputTypesForNode()
448 Status InOutTypesForNode(const NodeDef& node_def, const OpDef& op_def, in InOutTypesForNode()
454 Status NumOutputsForNode(const NodeDef& node_def, const OpDef& op_def, in NumOutputsForNode()
462 Status ValidateNodeDef(const NodeDef& node_def, const OpDef& op_def) { in ValidateNodeDef()
489 std::unordered_map<string, const OpDef::AttrDef*> op_attrs; in ValidateNodeDef()
558 Status ComputeArgRange(const NodeDef& node_def, const OpDef::ArgDef& arg_def, in ComputeArgRange()
[all …]
Dop_compatibility_test.cc44 const OpDef* RegisteredOpDef() { in RegisteredOpDef()
45 const OpDef* op_def; in RegisteredOpDef()
50 void ExpectSuccess(const OpDef& old_op_def) { in ExpectSuccess()
57 const OpDef* new_op_def = RegisteredOpDef(); in ExpectSuccess()
92 void ExpectIncompatible(const OpDef& old_op_def, const OpDef& new_op_def, in ExpectIncompatible()
105 void ExpectInvalid(const OpDef& old_op_def, const string& validation_error, in ExpectInvalid()
113 const OpDef* new_op_def = RegisteredOpDef(); in ExpectInvalid()
129 void ExpectTypeMismatch(const OpDef& old_op_def, in ExpectTypeMismatch()
137 const OpDef* new_op_def = RegisteredOpDef(); in ExpectTypeMismatch()
155 void ExpectRenameFailure(const OpDef& old_op_def, in ExpectRenameFailure()
[all …]
Dop_def_builder.cc146 void FinalizeAttr(StringPiece spec, OpDef* op_def, in FinalizeAttr()
148 OpDef::AttrDef* attr = op_def->add_attr(); in FinalizeAttr()
336 void FinalizeInputOrOutput(StringPiece spec, bool is_output, OpDef* op_def, in FinalizeInputOrOutput()
338 OpDef::ArgDef* arg = in FinalizeInputOrOutput()
367 const OpDef::AttrDef* attr = FindAttr(type_or_attr, *op_def); in FinalizeInputOrOutput()
392 OpDef::AttrDef* attr = FindAttrMutable(arg->number_attr(), op_def); in FinalizeInputOrOutput()
400 OpDef::AttrDef* attr = FindAttrMutable(arg->type_list_attr(), op_def); in FinalizeInputOrOutput()
425 void FinalizeControlOutput(StringPiece name, OpDef* op_def, in FinalizeControlOutput()
462 void FinalizeDoc(const string& text, OpDef* op_def, in FinalizeDoc()
639 OpDef* op_def = &op_reg_data->op_def; in Finalize()
Dop_def_builder_test.cc31 static void CanonicalizeAttrTypeListOrder(OpDef* def) { in CanonicalizeAttrTypeListOrder()
48 OpDef& op_def = op_reg_data.op_def; in ExpectSuccess()
50 OpDef expected; in ExpectSuccess()
68 OpDef& op_def = op_reg_data.op_def; in ExpectOrdered()
70 OpDef expected; in ExpectOrdered()
632 const OpDef& op_def = op_reg_data.op_def; in TEST_F()
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dfusion_utils.h34 const OpDef& first_function_signature,
35 const OpDef& second_function_signature, OpDef* fused_function_signature)>;
67 bool CanCompose(const OpDef& first_signature, const OpDef& second_signature);
69 void ComposeSignature(const OpDef& first_signature,
70 const OpDef& second_signature, OpDef* fused_signature);
84 void CombineSignature(const OpDef& first_signature,
85 const OpDef& second_signature, OpDef* fused_signature);
96 bool HasSameSignature(const OpDef& first_signature,
97 const OpDef& second_signature);
100 void SameSignature(const OpDef& first_signature, const OpDef& second_signature,
[all …]
Dfusion_utils.cc151 OpDef GetUniqueSignature(const OpDef& first_signature, in GetUniqueSignature()
152 const OpDef& second_signature, in GetUniqueSignature()
157 OpDef signature; in GetUniqueSignature()
287 void CheckIfCanCompose(const OpDef& first_signature, in CheckIfCanCompose()
288 const OpDef& second_signature) { in CheckIfCanCompose()
306 bool CanCompose(const OpDef& first_signature, const OpDef& second_signature) { in CanCompose()
320 void ComposeSignature(const OpDef& first_signature, in ComposeSignature()
321 const OpDef& second_signature, OpDef* fused_signature) { in ComposeSignature()
336 void CombineSignature(const OpDef& first_signature, in CombineSignature()
337 const OpDef& second_signature, OpDef* fused_signature) { in CombineSignature()
[all …]
/external/tensorflow/tensorflow/js/ops/
Dts_op_gen.cc34 ArgDefs(const OpDef::ArgDef& op_def_arg, const ApiDef::Arg& api_def_arg) in ArgDefs()
37 const OpDef::ArgDef& op_def_arg;
43 OpAttrs(const OpDef::AttrDef& op_def_attr, const ApiDef::Attr& api_def_attr) in OpAttrs()
46 const OpDef::AttrDef& op_def_attr;
53 GenTypeScriptOp(const OpDef& op_def, const ApiDef& api_def);
63 string InputForAttr(const OpDef::AttrDef& op_def_attr);
69 const OpDef& op_def_;
89 GenTypeScriptOp::GenTypeScriptOp(const OpDef& op_def, const ApiDef& api_def) in GenTypeScriptOp()
155 string GenTypeScriptOp::InputForAttr(const OpDef::AttrDef& op_def_attr) { in InputForAttr()
229 void WriteTSOp(const OpDef& op_def, const ApiDef& api_def, WritableFile* ts) { in WriteTSOp()
/external/tensorflow/tensorflow/core/api_def/
Dupdate_api_def.cc41 void FillBaseApiDef(ApiDef* api_def, const OpDef& op) { in FillBaseApiDef()
72 bool OpHasDocs(const OpDef& op) { in OpHasDocs()
96 bool CheckDocsMatch(const OpDef& op1, const OpDef& op2) { in CheckDocsMatch()
125 bool ValidateOpDocs(const OpDef& op, const string& doc) { in ValidateOpDocs()
147 string RemoveDoc(const OpDef& op, const string& file_contents, in RemoveDoc()
189 void RemoveDocs(const std::vector<const OpDef*>& ops, in RemoveDocs()
229 string CreateApiDef(const OpDef& op) { in CreateApiDef()
244 std::vector<const OpDef*> new_ops_with_docs; in CreateApiDefs()
Dupdate_api_def.h26 string CreateApiDef(const OpDef& op);
32 string RemoveDoc(const OpDef& op, const string& file_contents,
Dupdate_api_def_test.cc70 OpDef op; in TEST()
136 OpDef op1, op2, op3; in TEST()
173 OpDef op; in TEST()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dresource_operation_table_test.cc27 bool IsResourceArgDef(const OpDef::ArgDef& arg_def) { in IsResourceArgDef()
31 bool HasResourceInputOrOutput(const OpDef& op_def) { in HasResourceInputOrOutput()
46 const OpDef* op_def; in TEST()
/external/tensorflow/tensorflow/core/grappler/
Dgraph_view.cc25 const protobuf::RepeatedPtrField<OpDef::ArgDef>& args, in OpPortIdToArgId()
58 int OpOutputPortIdToArgId(const NodeDef& node, const OpDef& op, int port_id) { in OpOutputPortIdToArgId()
62 int OpInputPortIdToArgId(const NodeDef& node, const OpDef& op, int port_id) { in OpInputPortIdToArgId()
/external/tensorflow/tensorflow/core/grappler/utils/
Dfunctions.cc295 const OpDef::ArgDef& arg, DataType* data_type) const { in GetArgType()
397 const auto is_type_parametrized = [](const OpDef::ArgDef& arg) { in HasParametrizedType()
432 const auto resolve_type_attr = [&](const OpDef::ArgDef& arg) { in InstantiationTypeParameters()
486 const OpDef& signature = func.signature(); in MakeGrapplerFunctionItem()
494 for (const OpDef::AttrDef& attr : signature.attr()) { in MakeGrapplerFunctionItem()
524 for (const OpDef::ArgDef& input : signature.input_arg()) { in MakeGrapplerFunctionItem()
531 for (const OpDef::ArgDef& output : signature.output_arg()) { in MakeGrapplerFunctionItem()
543 for (const OpDef::ArgDef& input : signature.input_arg()) { in MakeGrapplerFunctionItem()
567 const auto output_node_name = [&func_body_nodes](const OpDef::ArgDef& out, in MakeGrapplerFunctionItem()
606 for (const OpDef::ArgDef& out : signature.output_arg()) { in MakeGrapplerFunctionItem()
[all …]
/external/tensorflow/tensorflow/compiler/jit/
Dextract_outside_compilation_pass_test.cc306 [&](const string &op, const OpDef **sig) { in TEST_F()
353 [&](const string &op, const OpDef **sig) { in TEST_F()
438 [&](const string &op, const OpDef **sig) { in TEST_F()
482 [&](const string &op, const OpDef **sig) { in TEST_F()
499 [&](const string &op, const OpDef **sig) { in TEST_F()
610 [&](const string &op, const OpDef **sig) { in TEST_F()
660 [&](const string &op, const OpDef **sig) { in TEST_F()
769 [&](const string &op, const OpDef **sig) { in TEST_F()
909 [&](const string &op, const OpDef **sig) { in TEST_F()
924 [&](const string &op, const OpDef **sig) { in TEST_F()
[all …]
/external/tensorflow/tensorflow/python/framework/
Dpython_op_gen_internal.h71 GenPythonOp(const OpDef& op_def, const ApiDef& api_def,
96 const OpDef& op_def_;

123456