Searched defs:select_scatter (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_cuda_repro.py | 923 def fn(arg7_1, add_1, permute_2, select_scatter, slice_8): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/ |
| H A D | hlo_function_importer.cc | 822 auto select_scatter = Cast<HloSelectAndScatterInstruction>(instruction); in ImportInstructionImpl() local
|
| /aosp_15_r20/external/pytorch/torch/_refs/ |
| H A D | __init__.py | 6222 def select_scatter(x: TensorLikeType, src: TensorLikeType, dim: int, index: int): function
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | lowering.py | 2493 def select_scatter(x, src, dim: int, index: int): function
|