Searched defs:output_axis (Results 1 – 3 of 3) 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
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()