Searched refs:inner_rank (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | conv_utils.py | 471 def squeeze_batch_dims(inp, op, inner_rank): argument
491 inner_shape = shape[-inner_rank:]
493 inner_shape = array_ops.shape(inp)[-inner_rank:]
495 batch_shape = shape[:-inner_rank]
497 batch_shape = array_ops.shape(inp)[:-inner_rank]
507 out_inner_shape = out_reshaped.shape[-inner_rank:]
509 out_inner_shape = array_ops.shape(out_reshaped)[-inner_rank:]
514 out.set_shape(inp.shape[:-inner_rank] + out.shape[-inner_rank:])
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | dynamic_ragged_shape.py | 62 if spec.inner_rank == 0:
723 inner_rank = self.inner_rank
724 if inner_rank is None:
727 return self.num_row_partitions + inner_rank
739 def inner_rank(self): member in DynamicRaggedShape
757 elif self.inner_rank == 0:
759 elif new_inner_rank == self.inner_rank:
761 elif new_inner_rank < self.inner_rank:
773 assert new_inner_rank > self.inner_rank
774 new_dimensions = new_inner_rank - self.inner_rank
[all …]
|
| D | dynamic_ragged_shape_test.py | 1278 self.assertAllEqual(actual.inner_rank, dense_rank)
3268 self.assertEqual(spec.inner_rank, 4)
3511 actual = spec.inner_rank
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_ops.py | 428 def squeeze_batch_dims(inp, op, inner_rank, name=None): argument
450 inner_shape = shape[-inner_rank:]
452 inner_shape = array_ops.shape(inp)[-inner_rank:]
454 batch_shape = shape[:-inner_rank]
456 batch_shape = array_ops.shape(inp)[:-inner_rank]
466 out_inner_shape = out_reshaped.shape[-inner_rank:]
468 out_inner_shape = array_ops.shape(out_reshaped)[-inner_rank:]
473 out.set_shape(inp.shape[:-inner_rank] + out.shape[-inner_rank:])
2088 inner_rank=3,
2479 inner_rank=3,
[all …]
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.experimental.-dynamic-ragged-shape.pbtxt | 17 name: "inner_rank"
|
| D | tensorflow.experimental.-dynamic-ragged-shape.-spec.pbtxt | 16 name: "inner_rank"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.experimental.-dynamic-ragged-shape.pbtxt | 17 name: "inner_rank"
|
| D | tensorflow.experimental.-dynamic-ragged-shape.-spec.pbtxt | 16 name: "inner_rank"
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | convolutional.py | 269 outputs, _apply_fn, inner_rank=self.rank + 1)
|