- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for ReversePermutation (0.19 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_layout_helper.cc
#include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops_layout_helper.h" #include "mlir/Support/LLVM.h" // from @llvm-project namespace mlir { namespace TF { SmallVector<int64_t, 4> ReversePermutation(ArrayRef<int64_t> permutation) { SmallVector<int64_t, 4> reverse(permutation.size()); for (size_t i = 0; i < permutation.size(); ++i) { reverse[permutation[i]] = i; } return reverse; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
// Use new paddings. setOperand(1, shuffled_paddings_op); // Change the result type. getResult().setType(ShuffleRankedTensorType(getResult().getType(), ReversePermutation(permutation)) .cast<TensorType>()); return success(); } //===----------------------------------------------------------------------===// // ParseExampleV2Op
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0)