Searched refs:output_shape_tensor (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/contrib/distributions/python/ops/bijectors/ |
| D | fill_triangular.py | 134 def _inverse_event_shape_tensor(self, output_shape_tensor): argument
135 batch_shape, n = output_shape_tensor[:-2], output_shape_tensor[-1]
138 n, output_shape_tensor[-2], message="Matrix must be square.")
141 d = math_ops.cast(n * (n + 1) / 2, output_shape_tensor.dtype)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | sparse_ops.py | 1775 output_shape_tensor = math_ops.maximum(
1779 output_shape_tensor = ops.convert_to_tensor(new_shape)
1780 output_shape_tensor.get_shape().assert_has_rank(1)
1781 output_shape_tensor = math_ops.cast(output_shape_tensor, dtypes.int64)
1784 if output_shape_tensor.get_shape().rank is not None:
1785 output_shape_tensor.get_shape().dims[0].merge_with(
1788 output_shape_tensor_const = tensor_util.constant_value(output_shape_tensor)
1798 output_shape_tensor = output_shape_tensor_const
1801 output_shape_tensor = control_flow_ops.with_dependencies([
1803 array_ops.shape(in_shape), array_ops.shape(output_shape_tensor))
[all …]
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | convolutional.py | 813 output_shape_tensor = array_ops.stack(output_shape)
817 output_shape_tensor,
1096 output_shape_tensor = array_ops.stack(output_shape)
1100 output_shape_tensor,
|