Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
Dlegalize_mhlo_to_thlo.cc156 DenseI64ArrayAttr knownExpandingDims; in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Transforms/
Dsymbolic_shape_optimization.cc142 llvm::SmallSetVector<int64_t, 4> *knownExpandingDims, in analyzeDynamicBroadcastInDimExpandingBehavior()
175 llvm::SmallSetVector<int64_t, 4> knownExpandingDims, knownNonexpandingDims; in matchAndRewrite() local