Searched refs:max_sum_shape (Results 1 – 2 of 2) sorted by relevance
76 TensorShape max_sum_shape = tmp_input_info.tensor_shape(); in configure() local77 _max_info = tmp_input_info.clone()->set_tensor_shape(max_sum_shape); in configure()78 …_sum_info = tmp_input_info.clone()->set_tensor_shape(max_sum_shape).set_data_type(… in configure()122 TensorShape max_sum_shape = src.tensor_shape(); in validate() local123 max_sum_shape.set(0, 1); in validate()124 TensorInfo tensor_info_max(src.clone()->set_tensor_shape(max_sum_shape).set_is_resizable(true)); in validate()125 …TensorInfo tensor_info_sum(src.clone()->set_tensor_shape(max_sum_shape).set_data_type(tmp_data_typ… in validate()
80 TensorShape max_sum_shape = tmp_input->tensor_shape(); in configure() local81 max_sum_shape.set(0, 1); in configure()85 TensorInfo max_info(tmp_input->clone()->set_tensor_shape(max_sum_shape)); in configure()132 TensorShape max_sum_shape = src->tensor_shape(); in validate() local133 max_sum_shape.set(0, 1); in validate()134 …const TensorInfo tensor_info_max_sum(src->clone()->set_tensor_shape(max_sum_shape).set_data_type(t… in validate()