Searched refs:squeeze_dims_ (Results 1 – 2 of 2) sorted by relevance
165 squeeze_dims_.insert(squeeze_dims.begin(), squeeze_dims.end()); in SqueezeOp()175 wrapped_squeeze_dims.reserve(squeeze_dims_.size()); in Compile()177 for (int32 dim : squeeze_dims_) { in Compile()216 std::unordered_set<int32> squeeze_dims_; member in tensorflow::__anon812f91ee0111::SqueezeOp
193 squeeze_dims_.insert(squeeze_dims.begin(), squeeze_dims.end()); in SqueezeOp()205 wrapped_squeeze_dims.reserve(squeeze_dims_.size()); in Compute()207 for (int32 dim : squeeze_dims_) { in Compute()259 std::unordered_set<int32> squeeze_dims_;