Searched refs:flattened_shape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | core.py | 673 flattened_shape = constant_op.constant([inputs.shape[0], -1])
674 return array_ops.reshape(inputs, flattened_shape)
686 flattened_shape = constant_op.constant([-1, last_dim])
688 flattened_shape = constant_op.constant([int(batch_dim), -1])
690 flattened_shape = [array_ops.shape_v2(inputs)[0], -1]
691 return array_ops.reshape(inputs, flattened_shape)
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | while_loop_simplifier.cc | 882 Shape flattened_shape = ShapeUtil::MakeTupleShape(flattened_shape_elems); in TryFlattenNestedTuples() local
918 0, flattened_shape, in TryFlattenNestedTuples()
930 0, flattened_shape, in TryFlattenNestedTuples()
943 flattened_shape, in TryFlattenNestedTuples()
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | pfor.py | 2666 flattened_shape = array_ops.concat([[-1], shape[2:]], axis=0)
2667 t = array_ops.reshape(t, flattened_shape)
|