Searched refs:GetDataFormatPermutation (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| H A D | tf_ops_layout_helper.h | 36 SmallVector<int64_t, 4> GetDataFormatPermutation(StringRef from, StringRef to);
55 auto perm = GetDataFormatPermutation(op->data_format(), data_format); in UpdateDataFormat()
108 GetDataFormatPermutation(op->data_format(), target_data_format);
|
| H A D | tf_ops_layout_helper.cc | 29 SmallVector<int64_t, 4> GetDataFormatPermutation(StringRef from, StringRef to) { in GetDataFormatPermutation() function
|
| H A D | tf_ops_a_m.cc | 1706 auto perm = GetDataFormatPermutation(this->data_format(), data_format); in UpdateDataFormat()
1878 auto perm = GetDataFormatPermutation(src_data_format, data_format); in UpdateDataFormat()
1952 auto perm = GetDataFormatPermutation(src_data_format, data_format); in UpdateDataFormat()
3141 auto perm = GetDataFormatPermutation(src_data_format, new_data_format); in UpdateDataFormat()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | layout_optimization.cc | 129 GetDataFormatPermutation(data_format, target_data_format); in runOnOperation()
133 GetDataFormatPermutation(target_data_format, data_format); in runOnOperation()
|