Searched refs:input_weight_output_mask (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/aten/src/ATen/native/ |
| D | Convolution.cpp | 2054 std::array<bool, 2> input_weight_output_mask = {output_mask[0], output_mask[1]}; in convolution_backward() local
2057 … weight, kernel_size, params.stride, params.padding, params.dilation, input_weight_output_mask); in convolution_backward()
2070 std::array<bool, 2> input_weight_output_mask = {output_mask[0], output_mask[1]}; in convolution_backward() local
2077 input_weight_output_mask); in convolution_backward()
2096 std::array<bool, 2> input_weight_output_mask = {output_mask[0], output_mask[1]}; in convolution_backward() local
2101 params.stride, params.dilation, params.groups, input_weight_output_mask); in convolution_backward()
2110 std::array<bool, 2> input_weight_output_mask = {output_mask[0], output_mask[1]}; in convolution_backward() local
2117 input_weight_output_mask); in convolution_backward()
|