Searched refs:before_dim_size_stack (Results 1 – 1 of 1) sorted by relevance
273 std::vector<int64> before_dim_size_stack; in AddInstruction() local276 before_dim_size_stack.push_back(hlo->operand(0)->shape().dimensions(i)); in AddInstruction()281 while (!before_dim_size_stack.empty() && !after_dim_size_stack.empty()) { in AddInstruction()282 int64 before_size = before_dim_size_stack.back(); in AddInstruction()285 hlo->operand(0)->shape().rank() - before_dim_size_stack.size(); in AddInstruction()288 before_dim_size_stack.pop_back(); in AddInstruction()298 before_dim_size_stack.push_back(before_size / after_size); in AddInstruction()310 if (!before_dim_size_stack.empty() || !after_dim_size_stack.empty()) { in AddInstruction()