Home
last modified time | relevance | path

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

/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DBatchMatmul.cpp86 Shape transposedShape = swapRowColumnDims(inputShape); in transposeRowsColumns() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
H A Dlower_general_dot.cc77 llvm::SmallVector<int64_t, 5> transposedShape; in transposeReshape() local
/aosp_15_r20/packages/modules/NeuralNetworks/common/include/
DOperationsExecutionUtils.h289 inline bool transposeFirstTwoDimensions(const Shape& shape, Shape* transposedShape) { in transposeFirstTwoDimensions()