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