Searched defs:output_axis (Results 1 – 5 of 5) sorted by relevance
63 for (int output_axis = 0; output_axis < shape.size(); ++output_axis) { in LocalTypeFromGlobalType() local89 for (int output_axis = 0; output_axis < shape.size(); ++output_axis) in GlobalTypeFromLocalType() local
14 def checkChannelsEqualized(self, tensor1, tensor2, output_axis, input_axis): argument
116 def cross_layer_equalization(module1, module2, output_axis=0, input_axis=1): argument
1912 const int output_axis = is_conv2d_backprop_input ? 2 : 3; in ConvertConv2DHelper() local2924 const int output_axis = is_conv3d_backprop_input ? 1 : 0; in ConvertConv3DHelper() local
3273 int output_axis, std::vector<uint64_t> *input_indices, in ComputePermutation()