Searched refs:with_rank_at_least (Results 1 – 25 of 41) sorted by relevance
12
172 cdbs = components_distribution.batch_shape.with_rank_at_least(1)[:-1]192 mixture_distribution.logits.shape.with_rank_at_least(1)[-1])194 components_distribution.batch_shape.with_rank_at_least(1)[-1])237 return self.components_distribution.batch_shape.with_rank_at_least(1)[:-1]
170 batch_shape.shape.with_rank_at_least(1)[0])188 batch_shape.shape.with_rank_at_least(1)[0])247 batch_shape.shape.with_rank_at_least(1)[0])
185 bs = dist.batch_shape.with_rank_at_least(1)605 stride = self.grid.shape.with_rank_at_least(794 deg = tensor_shape.dimension_value(p.shape.with_rank_at_least(1)[-1])833 if param.shape.with_rank_at_least(1)[-1] is not None:910 deg = tensor_shape.dimension_value(grid.shape.with_rank_at_least(1)[-1])944 deg = tensor_shape.dimension_value(grid.shape.with_rank_at_least(1)[-1])
133 logits_shape_static = self._logits.get_shape().with_rank_at_least(1)181 return self.logits.get_shape().with_rank_at_least(1)[-1:]
192 logits_shape_static = self._logits.get_shape().with_rank_at_least(1)246 return self.logits.get_shape().with_rank_at_least(1)[-1:]
152 labels_batch_shape = labels.shape.with_rank_at_least(1)[:-1]
242 x.shape.with_rank_at_least(1)[-1])266 x.shape.with_rank_at_least(1)[-1])412 inputs.shape.with_rank_at_least(1)[-1])528 x.shape.with_rank_at_least(1)[-1])
50 return tensor_shape.dimension_value(shape.shape.with_rank_at_least(1)[0])220 and x.shape.with_rank_at_least(event_ndims_in_)[
190 x.shape.with_rank_at_least(1)[-1])
272 elems_flat[0].get_shape().with_rank_at_least(1)[0]))275 elem.get_shape().with_rank_at_least(1)[0])))
355 a_shape = a.get_shape().with_rank_at_least(2)371 grad_u_shape = grad_u.get_shape().with_rank_at_least(2)372 grad_v_shape = grad_v.get_shape().with_rank_at_least(2)
737 inputs_got_shape = tuple(input_.get_shape().with_rank_at_least(3)1312 input_shape = first_input.get_shape().with_rank_at_least(2)1317 input_shape = flat_input.get_shape().with_rank_at_least(2)1327 fixed_batch_size = first_input.get_shape().with_rank_at_least(1)[0]
480 elems_flat[0].get_shape().with_rank_at_least(1)[0]))483 elem.get_shape().with_rank_at_least(1)[0])))
350 x_shape_static = x.get_shape().with_rank_at_least(1)835 x.shape.with_rank_at_least(1)[-1]) is not None:852 static_final_shape = x.shape.with_rank_at_least(1)[:-1].concatenate(948 x.shape.with_rank_at_least(2)[-1]) is not None:955 static_final_shape = x.shape.with_rank_at_least(2)[:-2].concatenate(1203 x.shape.with_rank_at_least(np.abs(axis))[axis])1253 return tensor_shape.dimension_value(x.shape.with_rank_at_least(1)[-1])
238 return self._mean_val.get_shape().with_rank_at_least(1)[:-1]244 return self._mean_val.get_shape().with_rank_at_least(1)[-1:]
259 return self.concentration.get_shape().with_rank_at_least(1)[-1:]
449 y.get_shape().with_rank_at_least(1)[:-event_ndims],475 y.get_shape().with_rank_at_least(1)[:-event_ndims],
71 signals.shape.with_rank_at_least(1)225 stfts.shape.with_rank_at_least(2)
118 input_tensor.shape.with_rank_at_least(fft_rank)137 input_tensor.shape.with_rank_at_least(fft_rank)
56 signal.shape.with_rank_at_least(2)
104 signal.shape.with_rank_at_least(1)
304 shape = tensor.get_shape().with_rank_at_least(1)345 shape = tensor.get_shape().with_rank_at_least(1)
247 values.shape.with_rank_at_least(1)1010 new_values.shape.with_rank_at_least(1)1116 tensor.shape.with_rank_at_least(ragged_rank + 1)1189 tensor.shape.with_rank_at_least(len(lengths) + 1)
74 name: "with_rank_at_least"