Searched refs:input_gen_fns (Results 1 – 4 of 4) sorted by relevance
| /external/pytorch/torch/_inductor/ |
| D | mkldnn_lowerings.py | 217 input_gen_fns = {
225 input_gen_fns=input_gen_fns,
279 input_gen_fns = {
287 input_gen_fns=input_gen_fns,
703 input_gen_fns = {
716 input_gen_fns=input_gen_fns,
1005 input_gen_fns = {
1011 input_gen_fns[7] = lambda x: V.graph.constants[x.get_name()] # For bias
1019 input_gen_fns=input_gen_fns,
1070 input_gen_fns = {
[all …]
|
| D | select_algorithm.py | 1181 input_gen_fns: Optional[Dict[int, Callable[[ir.Buffer], torch.Tensor]]] = None,
1190 if input_gen_fns is not None or layout.device.type == "cpu":
1222 return self.make_benchmark_fn(choices, input_nodes, layout, input_gen_fns)
1402 input_gen_fns=None, argument
1404 if input_gen_fns is None:
1405 input_gen_fns = {}
1410 x.get_name(): input_gen_fns.get(i, cls.benchmark_example_value)(x)
|
| /external/pytorch/torch/_inductor/kernel/ |
| D | flex_attention.py | 888 input_gen_fns = {
900 input_gen_fns=input_gen_fns,
1821 input_gen_fns = {
1837 input_gen_fns=input_gen_fns,
|
| D | flex_decoding.py | 528 input_gen_fns = {
540 input_gen_fns=input_gen_fns,
|