- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for GetDataFormatPermutation (0.22 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_layout_helper.h
namespace mlir { class MLIRContext; namespace TF { SmallVector<int64_t, 4> ReversePermutation(ArrayRef<int64_t> permutation); SmallVector<int64_t, 4> GetDataFormatPermutation(StringRef from, StringRef to); // Shuffle elements in the `attr` according to the permutation. Optional // `inner_size` allows to shuffle array attributes created from rank 2 tensors // on outer dimension only.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 08 01:19:25 UTC 2023 - 5.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_layout_helper.cc
SmallVector<int64_t, 4> reverse(permutation.size()); for (size_t i = 0; i < permutation.size(); ++i) { reverse[permutation[i]] = i; } return reverse; } SmallVector<int64_t, 4> GetDataFormatPermutation(StringRef from, StringRef to) { if (from == "NHWC" && to == "NCHW") { return {0, 3, 1, 2}; } else if (from == "NCHW" && to == "NHWC") { return {0, 2, 3, 1}; } else { return {}; } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.3K bytes - Viewed (0)