We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent c1fe854 commit f4af2b9Copy full SHA for f4af2b9
third_party/llga
torch_ipex/csrc/jit/codegen/onednn/kernel.cpp
@@ -70,10 +70,14 @@ ArgSpecs LlgaKernel::specializeInputSpecs(const TensorArgs &inputs) const {
70
71
ArgSpecs LlgaKernel::specializeOutputSpecs(const partition &partition,
72
const ArgSpecs &inputSpecs) const {
73
+ auto inputs = fmap(inputSpecs, toLogicalTensor);
74
+ auto outputs = fmap(graph_->outputs(), toLogicalTensor);
75
+ partition.infer_shape(inputs, outputs);
76
+
77
ArgSpecs outputSpecs;
78
outputSpecs.reserve(nOutputs_);
79
for (size_t i = 0; i < nOutputs_; i++) {
- auto spec = ArgSpec(graph_->outputs()[i]);
80
+ auto spec = ArgSpec(outputs[i]);
81
82
if (spec.is_quantized())
83
spec = getQuantizedSpec(spec, i);
0 commit comments