Searched refs:ir_node_to_tensor (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_inductor/kernel/ |
| H A D | conv.py | 394 ir.ir_node_to_tensor(x, guard_shape=True),
395 ir.ir_node_to_tensor(weight, guard_shape=True),
396 ir.ir_node_to_tensor(bias, guard_shape=True),
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | mkldnn_ir.py | 14 ir_node_to_tensor,
100 x_fake = ir_node_to_tensor(x, guard_shape=True)
101 weight_fake = ir_node_to_tensor(weight, guard_shape=True)
132 ir_node_to_tensor(bias, guard_shape=True) if bias is not None else bias
|
| H A D | ir.py | 225 def ir_node_to_tensor(x, guard_shape=True): function
4162 example_args.append(ir_node_to_tensor(x, guard_shape=True))
5856 x_fake = ir_node_to_tensor(x, guard_shape=True)
5857 weight_fake = ir_node_to_tensor(weight, guard_shape=True)
5888 ir_node_to_tensor(bias, guard_shape=True) if bias is not None else bias
|
| H A D | scheduler.py | 672 from .ir import ir_node_to_tensor
675 ir_node_to_tensor(input, guard_shape=False)
|