Home
last modified time | relevance | path

Searched refs:populateGatherToTorchIndexSelectPatterns (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
H A Dlegalize_gather_to_torch_index_select.cc137 populateGatherToTorchIndexSelectPatterns(&getContext(), &patterns); in runOnOperation()
145 void populateGatherToTorchIndexSelectPatterns(mlir::MLIRContext *context, in populateGatherToTorchIndexSelectPatterns() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/include/mlir-hlo/Dialect/mhlo/transforms/
H A Drewriters.h49 void populateGatherToTorchIndexSelectPatterns(mlir::MLIRContext *context,