Searched refs:GetStringAttr (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_util.cc | 34 absl::optional<string> GetStringAttr(const Node& n, const string& attr_name) { in GetStringAttr() function
77 GetStringAttr(*e->src(), outside_compilation_attr_name); in PreprocessControlEdgesBetweenOutsideCompilations()
79 GetStringAttr(*e->dst(), outside_compilation_attr_name); in PreprocessControlEdgesBetweenOutsideCompilations()
122 GetStringAttr(*e->src(), outside_compilation_attr_name); in PreprocessDataEdgesBetweenOutsideCompilations()
124 GetStringAttr(*e->dst(), outside_compilation_attr_name); in PreprocessDataEdgesBetweenOutsideCompilations()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | import_tensorflow.cc | 81 const string& GetStringAttr(const NodeDef& node, const string& attr_name) { in GetStringAttr() function
145 const string& value = GetStringAttr(node, attr_name); in CheckOptionalAttr()
678 const auto& padding = GetStringAttr(node, "padding"); in ConvertConvOperator()
701 CHECK_EQ(GetStringAttr(node, "data_format"), "NHWC"); in ConvertDepthwiseConvOperator()
751 const auto& padding = GetStringAttr(node, "padding"); in ConvertDepthwiseConvOperator()
1017 CHECK_EQ(GetStringAttr(node, "data_format"), "NHWC"); in ConvertMaxPoolOperator()
1039 const auto& padding = GetStringAttr(node, "padding"); in ConvertMaxPoolOperator()
1060 CHECK_EQ(GetStringAttr(node, "data_format"), "NHWC"); in ConvertAvgPoolOperator()
1078 const auto& padding = GetStringAttr(node, "padding"); in ConvertAvgPoolOperator()
1175 const auto mode = GetStringAttr(node, "mode"); in ConvertMirrorPadOperator()
[all …]
|