Searched refs:output_it (Results 1 – 2 of 2) sorted by relevance
49 for (auto output_it = output_dims.rbegin(); output_it != output_dims.rend(); in BroadcastTo() local50 ++output_it) { in BroadcastTo()52 if (!(*output_it == 0 && *input_it == 0) && in BroadcastTo()53 !(*input_it != 0 && *output_it % *input_it == 0)) { in BroadcastTo()61 if (*output_it == *input_it) { in BroadcastTo()62 broadcast_shape.push_back(*output_it); in BroadcastTo()63 } else if (*output_it != *input_it) { in BroadcastTo()68 broadcast_shape.push_back(*output_it / *input_it); in BroadcastTo()72 broadcast_shape.push_back(*output_it); in BroadcastTo()
713 auto output_it = src_output_idx_.find(inp.second); in input_shapes() local714 if (output_it == src_output_idx_.end()) continue; in input_shapes()719 const auto& output_shape = output_shapes.find(output_it->second); in input_shapes()