Searched refs:torch_index_select (Results 1 – 4 of 4) sorted by relevance
117 auto torch_index_select = rewriter.create<TorchIndexSelectOp>( in matchAndRewrite() local124 torch_index_select); in matchAndRewrite()
5 // CHECK: [[TIS:%.+]] = "mhlo.torch_index_select"(%arg0, %arg1) {18 // CHECK: [[TIS:%.+]] = "mhlo.torch_index_select"(%arg0, %arg1) {
1287 def HLO_TorchIndexSelectOp : HLO_Op<"torch_index_select", [NoSideEffect]> {
4112 …// CHECK: "mhlo.torch_index_select"(%arg0, %arg1) {batch_dims = 1 : i64, dim = 2 : i64} : (tensor<…4120 …// CHECK: "mhlo.torch_index_select"(%arg0, %arg1) {batch_dims = 1 : i64, dim = 2 : i64} : (tensor<…