Searched refs:min_max_shape (Results 1 – 2 of 2) sorted by relevance
106 min_max_shape = [input_shape[-1]]108 min_max_shape = []113 shape=min_max_shape,119 shape=min_max_shape,232 min_max_shape = [input_shape[-1]]234 min_max_shape = []239 shape=min_max_shape,245 shape=min_max_shape,
385 TensorShape min_max_shape({input.dim_size(input.dims() - 1)}); in Compute() local388 context->allocate_output(1, min_max_shape, &grad_wrt_min)); in Compute()392 context->allocate_output(2, min_max_shape, &grad_wrt_max)); in Compute()