Searched refs:num_squeeze_dims (Results 1 – 3 of 3) sorted by relevance
45 const int num_squeeze_dims = op_context.params->num_squeeze_dims; in Prepare() local56 if (num_squeeze_dims == 0) { in Prepare()63 for (int idx = 0; idx < num_squeeze_dims; ++idx) { in Prepare()
391 int num_squeeze_dims; member
1941 params->num_squeeze_dims = squeeze_dims->size(); in ParseSqueeze()1943 params->num_squeeze_dims = 0; in ParseSqueeze()
Completed in 13 milliseconds