Home
last modified time | relevance | path

Searched defs:transpose_shape (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
H A Dlstm_utils.cc101 auto transpose_shape = in Transpose() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dlower_tf.cc1138 SmallVector<int64_t> transpose_shape(permutation.size()); in matchAndRewrite() local
H A Dlegalize_hlo.cc2957 llvm::SmallVector<int64_t, 4> transpose_shape; in matchAndRewrite() local