Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
Dlegalize_to_linalg_utils.cc77 SmallVector<Value, 1> reifiedShapes; in getInitTensorFor() local
Dgroup_reduction_dimensions.cc158 llvm::SmallVector<Value, 1> reifiedShapes; in tryLowerTo1DOr2DReduction() local
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/IR/
Dhlo_ops.cc5938 llvm::SmallVector<Value> reifiedShapes; in matchAndRewrite() local
5980 llvm::SmallVector<Value> reifiedShapes; in matchAndRewrite() local