Searched refs:new_spatial_dim (Results 1 – 1 of 1) sorted by relevance
409 int64 new_spatial_dim = reshaped_activation_shape.dimensions().size() - 1; in HandleConvolution() local429 dim_numbers.add_input_spatial_dimensions(new_spatial_dim); in HandleConvolution()431 dim_numbers.set_kernel_input_feature_dimension(new_spatial_dim); in HandleConvolution()432 dim_numbers.add_output_spatial_dimensions(new_spatial_dim); in HandleConvolution()449 ShapeUtil::DeleteDimension(new_spatial_dim, new_convolution->shape()); in HandleConvolution()