Searched refs:wrapped_squeeze_dims (Results 1 – 2 of 2) sorted by relevance
174 std::unordered_set<int32> wrapped_squeeze_dims; in Compile() local175 wrapped_squeeze_dims.reserve(squeeze_dims_.size()); in Compile()187 wrapped_squeeze_dims.insert(dim); in Compile()194 if (!wrapped_squeeze_dims.empty()) { in Compile()195 if (wrapped_squeeze_dims.count(i) > 0) { in Compile()
204 std::unordered_set<int32> wrapped_squeeze_dims; in Compute() local205 wrapped_squeeze_dims.reserve(squeeze_dims_.size()); in Compute()218 wrapped_squeeze_dims.insert(dim); in Compute()225 if (!wrapped_squeeze_dims.empty()) { in Compute()226 if (wrapped_squeeze_dims.count(i) > 0) { in Compute()