Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlegalize_gather_to_torch_index_select.cc130 struct LegalizeGatherToTorchIndexSelectPass struct
131 : public PassWrapper<LegalizeGatherToTorchIndexSelectPass, FunctionPass> {
147 return std::make_unique<LegalizeGatherToTorchIndexSelectPass>(); in createLegalizeGatherToTorchIndexSelectPass()
/external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/transforms/
Dmhlo_passes.td42 def LegalizeGatherToTorchIndexSelectPass : Pass<"mhlo-legalize-gather-to-torch-index-select", "Func…