Searched refs:effective_num_dimensions (Results 1 – 1 of 1) sorted by relevance
337 const int effective_num_dimensions = std::max(2, num_dimensions); in GetGpuConvConfig() local375 input_descriptor = BatchDescriptor(effective_num_dimensions); in GetGpuConvConfig()383 static_cast<DimIndex>(effective_num_dimensions - dim - 1), in GetGpuConvConfig()388 filter_descriptor = FilterDescriptor(effective_num_dimensions); in GetGpuConvConfig()396 static_cast<DimIndex>(effective_num_dimensions - dim - 1), in GetGpuConvConfig()400 config.conv_desc = ConvolutionDescriptor(effective_num_dimensions); in GetGpuConvConfig()406 static_cast<DimIndex>(effective_num_dimensions - dim - 1), in GetGpuConvConfig()409 static_cast<DimIndex>(effective_num_dimensions - dim - 1), in GetGpuConvConfig()412 static_cast<DimIndex>(effective_num_dimensions - dim - 1), in GetGpuConvConfig()417 output_descriptor = BatchDescriptor(effective_num_dimensions); in GetGpuConvConfig()[all …]