Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlegalize_gather_to_torch_index_select.cc135 PopulateGatherToTorchIndexSelectPatterns(&getContext(), &patterns); in runOnFunction()
141 void PopulateGatherToTorchIndexSelectPatterns( in PopulateGatherToTorchIndexSelectPatterns() function
/external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/transforms/
Drewriters.h49 void PopulateGatherToTorchIndexSelectPatterns(