Searched refs:GetStringAttr (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_util.cc | 40 absl::optional<string> GetStringAttr(const Node& n, const string& attr_name) { in GetStringAttr() function
83 GetStringAttr(*e->src(), outside_compilation_attr_name); in PreprocessControlEdgesBetweenOutsideCompilations()
85 GetStringAttr(*e->dst(), outside_compilation_attr_name); in PreprocessControlEdgesBetweenOutsideCompilations()
128 GetStringAttr(*e->src(), outside_compilation_attr_name); in PreprocessDataEdgesBetweenOutsideCompilations()
130 GetStringAttr(*e->dst(), outside_compilation_attr_name); in PreprocessDataEdgesBetweenOutsideCompilations()
354 GetStringAttr(*e->src(), outside_compilation_attr_name); in OutsideCompilationClusterDependencies()
356 GetStringAttr(*e->dst(), outside_compilation_attr_name); in OutsideCompilationClusterDependencies()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | import_tensorflow.cc | 82 const std::string& GetStringAttr(const NodeDef& node, in GetStringAttr() function
148 const std::string& value = GetStringAttr(node, attr_name); in CheckOptionalAttr()
887 const auto& padding = GetStringAttr(node, "padding"); in ConvertConvOperator()
919 CHECK_EQ(GetStringAttr(node, "data_format"), "NHWC"); in ConvertDepthwiseConvOperator()
966 const auto& padding = GetStringAttr(node, "padding"); in ConvertDepthwiseConvOperator()
1272 CHECK_EQ(GetStringAttr(node, "data_format"), "NHWC"); in ConvertMaxPoolOperator()
1294 const auto& padding = GetStringAttr(node, "padding"); in ConvertMaxPoolOperator()
1315 CHECK_EQ(GetStringAttr(node, "data_format"), "NHWC"); in ConvertAvgPoolOperator()
1333 const auto& padding = GetStringAttr(node, "padding"); in ConvertAvgPoolOperator()
1430 const auto mode = GetStringAttr(node, "mode"); in ConvertMirrorPadOperator()
[all …]
|