Searched refs:dimension_at_index (Results 1 – 16 of 16) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | row_partition.py | 716 nsplits = tensor_shape.dimension_at_index(self._row_splits.shape, 0)
789 nrows = tensor_shape.dimension_at_index(self._row_splits.shape, 0) - 1
793 nrows = tensor_shape.dimension_at_index(self._row_lengths.shape, 0)
813 nvals = tensor_shape.dimension_at_index(self._value_rowids.shape, 0)
1134 [tensor_shape.dimension_at_index(self._nrows, 0) + 1])
|
| D | ragged_functional_ops.py | 163 nrows = tensor_shape.dimension_at_index(value.flat_values.shape, 0).value
|
| D | ragged_getitem.py | 137 nsplits = tensor_shape.dimension_at_index(inner_rt.row_splits.shape, 0)
260 nsplits = tensor_shape.dimension_at_index(inner_rt.row_splits.shape, 0)
|
| D | ragged_tensor.py | 1617 dim_len = tensor_shape.dimension_at_index(tensor.shape, axis).value
1688 const_nrows = tensor_shape.dimension_at_index(tensor.shape, 0).value
1689 const_ncols = tensor_shape.dimension_at_index(tensor.shape, 1).value
2358 outer_dim = tensor_shape.dimension_at_index(self._shape, 0)
2869 const_nvals = tensor_shape.dimension_at_index(values.shape, 0).value
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.compat.pbtxt | 36 name: "dimension_at_index"
|
| D | tensorflow.pbtxt | 1136 name: "dimension_at_index"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.compat.pbtxt | 36 name: "dimension_at_index"
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linear_operator.py | 680 tensor_shape.dimension_at_index(
725 tensor_shape.dimension_at_index(
873 tensor_shape.dimension_at_index(
928 tensor_shape.dimension_at_index(
|
| D | linear_operator_low_rank_update.py | 291 tensor_shape.dimension_at_index(uv_shape, -1).assert_is_compatible_with(
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | rnn.py | 1111 static_batch_size = tensor_shape.dimension_at_index(input_shape[0], 0)
1116 tensor_shape.dimension_at_index(input_shape_i, 0))
1340 batch_size, input_size = tensor_shape.dimension_at_index(
|
| D | sparse_ops.py | 428 max(tensor_shape.dimension_at_index(shape, dim)
433 tensor_shape.dimension_at_index(shape, axis)
1984 tensor_shape.dimension_at_index(retain_shape, 0))
|
| D | nn_ops.py | 1264 input_channels_dim = tensor_shape.dimension_at_index(
1268 input_channels_dim = tensor_shape.dimension_at_index(
1273 filter_dim = tensor_shape.dimension_at_index(filter_shape, num_spatial_dims)
|
| D | image_ops_impl.py | 1383 new_height_const = tensor_shape.dimension_at_index(size_const_as_shape,
1385 new_width_const = tensor_shape.dimension_at_index(size_const_as_shape,
|
| /external/tensorflow/tensorflow/python/ops/structured/ |
| D | structured_tensor.py | 237 static_nrows = tensor_shape.dimension_at_index(shape, 0)
1216 static_value_nrows = tensor_shape.dimension_at_index(value.shape, 0)
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | tensor_shape.py | 135 def dimension_at_index(shape, index): function
|
| /external/tensorflow/tensorflow/python/ops/distributions/ |
| D | util.py | 1330 tensor_shape.dimension_at_index(x.shape, axis) + count_ *
|