Searched refs:dimension_at_index (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linear_operator.py | 617 tensor_shape.dimension_at_index(
659 tensor_shape.dimension_at_index(
789 tensor_shape.dimension_at_index(
844 tensor_shape.dimension_at_index(
|
| D | linear_operator_low_rank_update.py | 284 tensor_shape.dimension_at_index(uv_shape, -1).assert_is_compatible_with(
|
| /external/tensorflow/tensorflow/contrib/training/python/training/ |
| D | sampling_ops.py | 307 tensor_shape.dimension_at_index(shape, 0).assert_is_compatible_with(
348 tensor_shape.dimension_at_index(shape, 0).assert_is_compatible_with(
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.compat.pbtxt | 36 name: "dimension_at_index"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.compat.pbtxt | 36 name: "dimension_at_index"
|
| D | tensorflow.pbtxt | 1100 name: "dimension_at_index"
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | rnn.py | 1097 static_batch_size = tensor_shape.dimension_at_index(input_shape[0], 0)
1102 tensor_shape.dimension_at_index(input_shape_i, 0))
1318 batch_size, input_size = tensor_shape.dimension_at_index(
|
| D | nn_ops.py | 1037 input_channels_dim = tensor_shape.dimension_at_index(
1041 input_channels_dim = tensor_shape.dimension_at_index(input_shape, 1)
|
| D | sparse_ops.py | 1698 tensor_shape.dimension_at_index(retain_shape, 0))
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | tensor_shape.py | 126 def dimension_at_index(shape, index): function
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_tensor.py | 649 nrows = tensor_shape.dimension_at_index(self._row_splits.shape, 0) - 1
|
| /external/tensorflow/tensorflow/python/ops/distributions/ |
| D | util.py | 1322 else (tensor_shape.dimension_at_index(
|