Searched refs:InputsKernel (Results 1 – 3 of 3) sorted by relevance
| /external/pytorch/torch/_inductor/ |
| D | mkldnn_lowerings.py | 513 ir.InputsKernel.unwrap_storage_for_input(w_zp),
531 ir.InputsKernel.unwrap_storage_for_input(x_zp),
536 ir.InputsKernel.unwrap_storage_for_input(w_zp),
777 ir.InputsKernel.unwrap_storage_for_input(w_zp),
810 ir.InputsKernel.unwrap_storage_for_input(x_zp),
815 ir.InputsKernel.unwrap_storage_for_input(w_zp),
|
| D | ir.py | 3521 self.inputs = InputsKernel.unwrap_storage(inputs)
3724 class InputsKernel(Buffer): class
3771 x = [InputsKernel.unwrap_storage_for_input(i) for i in x]
3773 x = InputsKernel.unwrap_storage_for_input(x)
3781 class NopKernel(InputsKernel):
3944 class ExternKernel(InputsKernel):
4719 t = InputsKernel.unwrap_storage_for_input(self.realize_input(v))
7496 InputsKernel,
7566 if isinstance(data, (InputsKernel, InputBuffer, ReinterpretView)):
|
| /external/pytorch/torch/_inductor/codegen/ |
| D | cpp_gemm_template.py | 683 template_buffer = ir.InputsKernel.unwrap_storage_for_input(output)
745 template_buffer.inputs[1] = ir.InputsKernel.unwrap_storage_for_input(
|