Home
last modified time | relevance | path

Searched defs:new_rhs_shape (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dcudnn_pad_for_convolutions.cc201 Shape new_rhs_shape = rhs->shape(); in TryResolvePaddedShapesForTensorCore() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
H A Dlegalize_tfl.cc1385 llvm::SmallVector<int64_t> new_rhs_shape{N}; in matchAndRewrite() local