Searched refs:src_data_format (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| H A D | tf_ops_a_m.cc | 1876 StringRef src_data_format = this->data_format(); in UpdateDataFormat() local
1878 auto perm = GetDataFormatPermutation(src_data_format, data_format); in UpdateDataFormat()
1893 getLoc(), filter_sizes(), StringAttr::get(getContext(), src_data_format), in UpdateDataFormat()
1950 StringRef src_data_format = this->data_format(); in UpdateDataFormat() local
1952 auto perm = GetDataFormatPermutation(src_data_format, data_format); in UpdateDataFormat()
1967 getLoc(), input_sizes(), StringAttr::get(getContext(), src_data_format), in UpdateDataFormat()
3139 StringRef src_data_format = data_format(); in UpdateDataFormat() local
3141 auto perm = GetDataFormatPermutation(src_data_format, new_data_format); in UpdateDataFormat()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | generic_layout_optimizer_transposer.cc | 74 absl::string_view src_data_format, in AttrDataFormatMatch() argument
78 return attr->s() == src_data_format; in AttrDataFormatMatch()
85 absl::string_view src_data_format) { in AttrDataFormatMatch() argument
87 return AttrDataFormatMatch(node, src_data_format, &missing); in AttrDataFormatMatch()
|