Searched defs:output_a (Results 1 – 4 of 4) sorted by relevance
46 std::vector<int>* output_a, in PairwiseVectorSelect()
421 const auto output_a = output.packed_accessor32<scalar_t, 4>(); in conv_depthwise2d_forward_out() local
597 auto output_a = output.accessor<scalar_t, 4>(); in slow_conv2d_forward_out_cpu() local
627 auto output_a = output.accessor<scalar_t, 5>(); in slow_conv3d_forward_out_cpu() local