Searched refs:codegen_reference (Results 1 – 4 of 4) sorted by relevance
| /external/pytorch/torch/_inductor/codegen/ |
| D | cpp_wrapper_cpu.py | 970 else x.codegen_reference(self.wrapper_call)
1435 (data,) = (t.codegen_reference() for t in node.inputs)
1867 src = inner_output.codegen_reference()
1895 conditional.predicate.codegen_reference(),
1900 predicate = conditional.predicate.codegen_reference()
1924 buf.codegen_reference() for buf in while_loop.carried_inputs
1927 buf.codegen_reference() for buf in while_loop.additional_inputs
2219 base_handle = raw_arg.codegen_reference()
2459 return val.codegen_reference()
|
| D | wrapper.py | 633 return [x.codegen_reference(self.wrapper_call) for x in V.graph.graph_outputs]
1133 (data,) = (t.codegen_reference() for t in node.inputs)
1781 return s.codegen_reference()
1963 outer_inputs = [buf.codegen_reference() for buf in conditional.operands]
1966 predicate = conditional.predicate.codegen_reference()
1984 buf.codegen_reference() for buf in while_loop.carried_inputs
1987 buf.codegen_reference() for buf in while_loop.additional_inputs
|
| /external/pytorch/torch/_inductor/ |
| D | ir.py | 346 def codegen_reference(self, writer=None): member in IRNode
2352 def codegen_reference(self, writer=None): member in ReinterpretView
3103 def codegen_reference(self, writer=None): member in Buffer
3191 def codegen_reference(self, writer=None): member in NoneAsConstantBuffer
3203 def codegen_reference(self, writer=None): member in ShapeAsConstantBuffer
4411 names = [i.codegen_reference() for i in x]
4412 codegen_reference = f'[{", ".join(names)}]'
4413 args.append(codegen_reference)
4426 args.append(x.codegen_reference())
4550 self.codegen_reference(),
[all …]
|
| D | mkldnn_ir.py | 618 args = [x.codegen_reference() for x in self.inputs]
857 args = [x.codegen_reference() for x in self.inputs]
1333 args = [x.codegen_reference() for x in self.inputs]
1551 args = [x.codegen_reference() for x in self.inputs]
|