Searched defs:output_axis (Results 1 – 2 of 2) sorted by relevance
2216 const int output_axis = is_conv2d_backprop_input ? 1 : 0; in ConvertConv2DHelper() local3178 const int output_axis = is_conv3d_backprop_input ? 1 : 0; in ConvertConv3DHelper() local
3068 int output_axis, std::vector<uint64_t> *input_indices, in ComputePermutation()