Home
last modified time | relevance | path

Searched refs:num_src_spatial_dims (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Dtensor_format.h601 const int num_src_spatial_dims = in ShapeFromFormat() local
603 std::vector<int64_t> spatial_dims(num_src_spatial_dims); in ShapeFromFormat()
604 for (int spatial_dim = 0; spatial_dim < num_src_spatial_dims; ++spatial_dim) { in ShapeFromFormat()
610 spatial_dims[num_src_spatial_dims - 1] *= 4; in ShapeFromFormat()