Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DConvolution.cpp2054 std::array<bool, 2> input_weight_output_mask = {output_mask[0], output_mask[1]}; in convolution_backward() local
2070 std::array<bool, 2> input_weight_output_mask = {output_mask[0], output_mask[1]}; in convolution_backward() local
2096 std::array<bool, 2> input_weight_output_mask = {output_mask[0], output_mask[1]}; in convolution_backward() local
2110 std::array<bool, 2> input_weight_output_mask = {output_mask[0], output_mask[1]}; in convolution_backward() local