Lines Matching refs:num_spatial_dims
57 int num_spatial_dims = GetTensorSpatialDims(input_rank, data_format_);
67 for (int i = 0; i < num_spatial_dims; ++i) {
71 for (int i = 0; i < num_spatial_dims; ++i) {
78 for (int i = 0; i < num_spatial_dims; ++i) {
80 transpose_order.push_back(i + 1 + num_spatial_dims);
82 transpose_order.push_back(feature_dim + num_spatial_dims);
85 for (int i = 0; i < num_spatial_dims; ++i) {
93 for (int i = 0; i < num_spatial_dims; ++i) {
98 for (int i = 0; i < num_spatial_dims; ++i) {
103 transpose_order.push_back(1 + num_spatial_dims);
104 for (int i = 0; i < num_spatial_dims; ++i) {
105 transpose_order.push_back(2 + num_spatial_dims + i);
111 for (int i = 0; i < num_spatial_dims; ++i) {