Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dtranspose_folding.cc131 const auto& transpose_dimensions = transpose.dimensions(); in FoldTransposeIntoConvolution() local
154 const auto& transpose_dimensions = transpose.dimensions(); in FoldTransposeIntoConvolution() local
H A Dalgebraic_simplifier.cc2094 std::vector<int64_t> transpose_dimensions(batch_dimensions.begin(), in NormalizeDotOperandToBatchMajorAndContractingMinor() local
5373 auto transpose_dimensions = arg->dimensions(); in HandleReduce() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dlegalize_hlo.cc2956 llvm::SmallVector<int64_t, 4> transpose_dimensions; in matchAndRewrite() local