Searched refs:full_axis_dim (Results 1 – 1 of 1) sorted by relevance
483 full_axis_dim = math_ops.reduce_max(first_dims)484 padded_input_tensor = _pad_util(input_tensor_t, full_axis_dim)492 start_pos = i * full_axis_dim698 def _pad_util(input_tensor, full_axis_dim): argument700 missing_axis_dim = full_axis_dim - array_ops.shape_v2(input_tensor)[0]