Home
last modified time | relevance | path

Searched defs:output_size_reshaped (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/aten/src/ATen/native/mkldnn/
DLinear.cpp207 std::vector<int64_t> output_size_reshaped = {input_reshaped.size(0), in mkldnn_linear_pointwise() local
294 std::vector<int64_t> output_size_reshaped = { in mkldnn_linear_pointwise_binary() local