Searched defs:normalized_shape_dim (Results 1 – 1 of 1) sorted by relevance
22 int64_t normalized_shape_dim) { in calc_out_mean_sizes()41 const auto normalized_shape_dim = graph->get_int_list(extra_args[0])->size(); in resize_native_layer_norm_node() local64 const auto normalized_shape_dim = in add_native_layer_norm_node() local