| /external/tensorflow/tensorflow/core/framework/ |
| D | op_def_util.h | 31 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 std::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 …]
|
| D | op_def_util.cc | 41 bool HasAttrStyleType(const OpDef::ArgDef& arg) { in HasAttrStyleType()
46 Status AllowedTypeValue(DataType dt, const OpDef::AttrDef& attr) { in AllowedTypeValue()
66 Status AllowedStringValue(const string& str, const OpDef::AttrDef& attr) { in AllowedStringValue()
89 const OpDef::AttrDef& attr) { in ValidateAttrValue()
151 const OpDef::AttrDef* FindAttr(StringPiece name, const OpDef& op_def) { in FindAttr()
160 OpDef::AttrDef* FindAttrMutable(StringPiece name, OpDef* op_def) { in FindAttrMutable()
169 const OpDef::ArgDef* FindInputArg(StringPiece name, const OpDef& op_def) { in FindInputArg()
195 static Status ValidateArg(const OpDef::ArgDef& arg, const OpDef& op_def, in ValidateArg()
204 const OpDef::AttrDef* attr = FindAttr(arg.number_attr(), op_def); in ValidateArg()
229 const OpDef::AttrDef* attr = FindAttr(arg.type_attr(), op_def); in ValidateArg()
[all …]
|
| D | op_def_util_test.cc | 30 OpDef FromText(const string& text) { in FromText()
31 OpDef op_def; in FromText()
36 OpDef::AttrDef ADef(const string& text) { in ADef()
37 OpDef::AttrDef attr_def; in ADef()
373 void ExpectDifferent(const OpDef::AttrDef& a1, const OpDef::AttrDef& a2) { in ExpectDifferent()
380 OpDef::AttrDef a = ADef( in TEST()
445 protobuf::RepeatedPtrField<OpDef::AttrDef> Rep( in Rep()
446 const std::vector<OpDef::AttrDef>& defs) { in Rep()
447 protobuf::RepeatedPtrField<OpDef::AttrDef> rep; in Rep()
448 for (const OpDef::AttrDef& def : defs) { in Rep()
[all …]
|
| D | node_def_builder.h | 37 typedef std::function<Status(const OpDef&, int, const NodeDef&,
74 NodeDefBuilder(StringPiece name, const OpDef* op_def);
144 const OpDef& op_def() const { return *op_def_; } in op_def()
152 const OpDef::ArgDef* NextArgDef();
159 void SingleInput(const OpDef::ArgDef* input_arg, StringPiece src_node,
161 void ListInput(const OpDef::ArgDef* input_arg,
168 void VerifyInputType(const OpDef::ArgDef* input_arg, DataType expected,
172 void VerifyInputRef(const OpDef::ArgDef* input_arg, DataType dt);
175 DataType MaybeAddRef(const OpDef::ArgDef* input_arg, DataType dt) { in MaybeAddRef()
184 const OpDef* op_def_;
|
| D | fake_input.cc | 31 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()
|
| D | node_def_util_test.cc | 34 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 …]
|
| D | graph_def_util.cc | 73 const OpDef* op_def; in AddDefaultAttrsToGraphDef()
89 const OpDef* producer_op_def; in RemoveNewDefaultAttrsFromNodeDef()
90 const OpDef* consumer_op_def; in RemoveNewDefaultAttrsFromNodeDef()
101 const OpDef::AttrDef* producer_attr_def = in RemoveNewDefaultAttrsFromNodeDef()
172 const OpDef* op_def; in StripDefaultAttributes()
184 for (const OpDef::AttrDef& attr_def : op_def->attr()) { in StripDefaultAttributes()
257 const OpDef* op_def; in StrippedOpListForGraph()
259 OpDef* stripped_op = stripped_op_list->add_op(); in StrippedOpListForGraph()
|
| D | op_def_builder.h | 42 OpRegistrationData(const OpDef& def) : op_def(def) {} in OpRegistrationData()
43 OpRegistrationData(const OpDef& def, const OpShapeInferenceFn& fn,
47 OpDef op_def;
166 OpDef* op_def() { return &op_reg_data_.op_def; } in op_def()
|
| D | node_def_util.h | 44 class OpDef; variable
322 Status InputTypeForNode(const NodeDef& node_def, const OpDef& op_def,
326 Status InputTypesForNode(const NodeDef& node_def, const OpDef& op_def,
330 Status OutputTypeForNode(const NodeDef& node_def, const OpDef& op_def,
334 Status OutputTypesForNode(const NodeDef& node_def, const OpDef& op_def,
336 Status OutputTypesForNode(const AttrSlice& attrs, const OpDef& op_def,
341 Status InOutTypesForNode(const NodeDef& node_def, const OpDef& op_def,
345 Status NumOutputsForNode(const NodeDef& node_def, const OpDef& op_def,
353 Status ValidateNodeDef(const NodeDef& node_def, const OpDef& op_def);
365 Status NameRangesForNode(const AttrSlice& attrs, const OpDef& op_def,
[all …]
|
| D | node_properties.h | 30 NodeProperties(const OpDef* op_def, NodeDef node_def, in NodeProperties()
36 NodeProperties(const OpDef* _op_def, NodeDef&& _node_def, in NodeProperties()
53 const OpDef* op_def; // not owned.
|
| D | node_def_builder.cc | 60 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()
|
| D | node_properties_test.cc | 25 OpDef ToOpDef(const OpDefBuilder& builder) { in ToOpDef()
55 const OpDef* get_op_def_addr() { return &op_reg_.op_def; } in get_op_def_addr()
61 void ValidateNodeProperties(const NodeProperties& props, const OpDef* op_def, in ValidateNodeProperties()
82 OpDef op_def; in TEST()
|
| D | node_def_util.cc | 446 const OpDef::ArgDef& arg_def, DataTypeVector* sig) { in AddArgToSig()
511 Status InputTypeForNode(const NodeDef& node_def, const OpDef& op_def, in InputTypeForNode()
527 Status InputTypesForNode(const NodeDef& node_def, const OpDef& op_def, in InputTypesForNode()
535 Status OutputTypeForNode(const NodeDef& node_def, const OpDef& op_def, in OutputTypeForNode()
551 Status OutputTypesForNode(const NodeDef& node_def, const OpDef& op_def, in OutputTypesForNode()
559 Status OutputTypesForNode(const AttrSlice& attrs, const OpDef& op_def, in OutputTypesForNode()
567 Status InOutTypesForNode(const NodeDef& node_def, const OpDef& op_def, in InOutTypesForNode()
573 Status NumOutputsForNode(const NodeDef& node_def, const OpDef& op_def, in NumOutputsForNode()
581 Status ValidateNodeDef(const NodeDef& node_def, const OpDef& op_def) { in ValidateNodeDef()
608 std::unordered_map<string, const OpDef::AttrDef*> op_attrs; in ValidateNodeDef()
[all …]
|
| D | op_compatibility_test.cc | 45 const OpDef* RegisteredOpDef() { in RegisteredOpDef()
46 const OpDef* op_def; in RegisteredOpDef()
51 void ExpectSuccess(const OpDef& old_op_def) { in ExpectSuccess()
58 const OpDef* new_op_def = RegisteredOpDef(); in ExpectSuccess()
93 void ExpectIncompatible(const OpDef& old_op_def, const OpDef& new_op_def, in ExpectIncompatible()
106 void ExpectInvalid(const OpDef& old_op_def, const string& validation_error, in ExpectInvalid()
114 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 …]
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | fusion_utils.h | 34 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 …]
|
| D | fusion_utils.cc | 152 OpDef GetUniqueSignature(const OpDef& first_signature, in GetUniqueSignature()
153 const OpDef& second_signature, in GetUniqueSignature()
158 OpDef signature; in GetUniqueSignature()
288 void CheckIfCanCompose(const OpDef& first_signature, in CheckIfCanCompose()
289 const OpDef& second_signature) { in CheckIfCanCompose()
307 bool CanCompose(const OpDef& first_signature, const OpDef& second_signature) { in CanCompose()
321 void ComposeSignature(const OpDef& first_signature, in ComposeSignature()
322 const OpDef& second_signature, OpDef* fused_signature) { in ComposeSignature()
337 void CombineSignature(const OpDef& first_signature, in CombineSignature()
338 const OpDef& second_signature, OpDef* fused_signature) { in CombineSignature()
[all …]
|
| D | map_and_filter_fusion.cc | 102 OpDef fused_sig = fused_map_func.signature(); in MakeFilterNode()
104 OpDef* sig = func->mutable_signature(); in MakeFilterNode()
109 OpDef::ArgDef* arg = sig->add_output_arg(); in MakeFilterNode()
144 OpDef fused_sig = fused_map_func.signature(); in MakeMapNode()
146 OpDef* sig = func->mutable_signature(); in MakeMapNode()
|
| /external/tensorflow/tensorflow/js/ops/ |
| D | ts_op_gen.cc | 34 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/ |
| D | update_api_def.cc | 41 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()
|
| D | update_api_def.h | 26 string CreateApiDef(const OpDef& op);
32 string RemoveDoc(const OpDef& op, const string& file_contents,
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | resource_operation_table_test.cc | 27 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/common_runtime/eager/ |
| D | eager_operation.h | 95 const tensorflow::OpDef* OpDef() const override { return op_def_; }; in OpDef() function
203 const tensorflow::OpDef* GetOpDef(Status* status);
214 void InferSingleTypeInputListAttrs(const OpDef::ArgDef& input_def,
216 void InferMixedTypeInputListAttrs(const OpDef::ArgDef& input_def,
258 const tensorflow::OpDef* op_def_; // op definition from protobuf
|
| D | attr_builder.cc | 57 Status OpDefForOp(const string& op_name, const OpDef** op_def) { in OpDefForOp()
83 const OpDef* op_def = nullptr; in AttrTypeMapForOp()
174 const OpDef* op_def = nullptr; in FillAttrValueMap()
190 bool ValueMatchesDefault(const OpDef* op_def, const string& attr_name, in ValueMatchesDefault()
194 for (const OpDef::AttrDef& attr_def : op_def->attr()) { in ValueMatchesDefault()
206 const OpDef* op_def = nullptr; in FillAttrValueMapWithoutDefaults()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | graph_view.cc | 25 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/common_runtime/ |
| D | function_def_utils.h | 30 class OpDef; variable
44 const std::function<Status(const string&, const OpDef**)>& get_func_sig,
|