Home
last modified time | relevance | path

Searched refs:with_rank_at_least (Results 1 – 25 of 25) sorted by relevance

/external/tensorflow/tensorflow/python/ops/signal/
Dspectral_ops.py75 signals.shape.with_rank_at_least(1)
228 stfts.shape.with_rank_at_least(2)
336 signals.shape.with_rank_at_least(1)
432 mdcts.shape.with_rank_at_least(2)
Dreconstruction_ops.py58 signal.shape.with_rank_at_least(2)
Dshape_ops.py124 signal.shape.with_rank_at_least(1)
Dfft_ops.py131 input_tensor.shape.with_rank_at_least(fft_rank)
155 input_tensor.shape.with_rank_at_least(fft_rank)
/external/tensorflow/tensorflow/python/ops/distributions/
Dutil.py361 x_shape_static = x.get_shape().with_rank_at_least(1)
844 x.shape.with_rank_at_least(1)[-1]) is not None:
861 static_final_shape = x.shape.with_rank_at_least(1)[:-1].concatenate(
954 x.shape.with_rank_at_least(2)[-1]) is not None:
961 static_final_shape = x.shape.with_rank_at_least(2)[:-2].concatenate(
1210 x.shape.with_rank_at_least(np.abs(axis))[axis])
1261 return tensor_shape.dimension_value(x.shape.with_rank_at_least(1)[-1])
Dmultinomial.py238 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:]
Ddirichlet_multinomial.py259 return self.concentration.get_shape().with_rank_at_least(1)[-1:]
Dtransformed_distribution.py449 y.get_shape().with_rank_at_least(1)[:-event_ndims],
475 y.get_shape().with_rank_at_least(1)[:-event_ndims],
Ddirichlet.py232 return self.concentration.get_shape().with_rank_at_least(1)[-1:]
Dcategorical.py206 logits_shape_static = self._logits.get_shape().with_rank_at_least(1)
/external/tensorflow/tensorflow/python/ops/
Dmap_fn.py438 elems_batchable[0].get_shape().with_rank_at_least(1)[0]))
443 tensor.get_shape().with_rank_at_least(1)[0])))
Drnn.py758 input_.get_shape().with_rank_at_least(3) for input_ in flat_input)
1334 input_shape = first_input.get_shape().with_rank_at_least(2)
1339 input_shape = flat_input.get_shape().with_rank_at_least(2)
1349 fixed_batch_size = first_input.get_shape().with_rank_at_least(1)[0]
Dfunctional_ops.py676 elems_flat[0].get_shape().with_rank_at_least(1)[0]))
681 elem.get_shape().with_rank_at_least(1)[0])))
Dlinalg_grad.py823 a_shape = a.get_shape().with_rank_at_least(2)
835 grad_u_shape = grad_u.get_shape().with_rank_at_least(2)
836 grad_v_shape = grad_v.get_shape().with_rank_at_least(2)
Dimage_ops_impl.py222 image_shape = image.get_shape().with_rank_at_least(3)
288 image_shape = image.get_shape().with_rank_at_least(2)
3997 shape1 = img1.get_shape().with_rank_at_least(3)
3998 shape2 = img2.get_shape().with_rank_at_least(3)
Ddata_flow_ops.py386 vals[0].get_shape().with_rank_at_least(1)[0])
390 val.get_shape().with_rank_at_least(1)[0])
Dnn_ops.py636 input_shape.with_rank_at_least(expected_input_rank)
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.-tensor-shape.pbtxt74 name: "with_rank_at_least"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.-tensor-shape.pbtxt74 name: "with_rank_at_least"
/external/tensorflow/tensorflow/python/eager/
Dbackprop.py1281 if not (target_shape.with_rank_at_least(2) and
1282 source.shape.with_rank_at_least(2) and
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_tensor.py287 values.shape.with_rank_at_least(1)
1323 new_values.shape.with_rank_at_least(1)
1568 tensor.shape.with_rank_at_least(ragged_rank + 1)
1586 tensor.shape.with_rank_at_least(len(lengths) + 1)
/external/tensorflow/tensorflow/python/framework/
Dtensor_shape.py1060 def with_rank_at_least(self, rank): member in TensorShape
/external/tensorflow/tensorflow/python/training/
Dinput.py674 shape_list = [s.with_rank_at_least(1)[1:] for s in shape_list]
/external/tensorflow/tensorflow/python/keras/layers/
Dcore.py1217 input_shape = input_shape.with_rank_at_least(2)
/external/tensorflow/tensorflow/python/keras/
Dbackend.py4330 input_.shape.with_rank_at_least(3)