Searched refs:as_shape (Results 1 – 25 of 36) sorted by relevance
12
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_sharding.py | 169 shape = tensor_shape.as_shape(shape)
189 return tensor_shape.as_shape(dims)
205 shape = tensor_shape.as_shape(shape)
217 return tensor_shape.as_shape(dims)
|
| D | tpu_feed.py | 275 tuple_shapes = [tensor_shape.as_shape(shape) for shape in tuple_shapes]
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | tensor_shape.py | 890 other = as_shape(other)
920 other = as_shape(other)
936 other = as_shape(other)
1050 other = as_shape(other)
1094 other = as_shape(other)
1145 other = as_shape(other)
1153 other = as_shape(other)
1169 def as_shape(shape): function
|
| D | tensor_shape_test.py | 368 tensor_shape.as_shape(proto))
374 tensor_shape.as_shape(proto))
376 partial_proto_shape = tensor_shape.as_shape(
|
| D | constant_op.py | 256 shape = tensor_shape.as_shape(shape)
|
| D | tensor_util.py | 533 tensor_shape=tensor_shape.as_shape(shape).as_proto())
843 return tensor_shape.as_shape(
|
| D | op_def_library.py | 199 return tensor_shape.as_shape(v).as_proto()
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | recurrent.py | 158 tensor_shape.as_shape(output_dim).as_list())
434 input_shape = tensor_shape.as_shape(input_shape)
450 output_dim = tensor_shape.as_shape(flat_output_size).as_list()
453 output_shape = tensor_shape.as_shape([time_step, batch] + output_dim)
455 output_shape = tensor_shape.as_shape([batch, time_step] + output_dim)
457 output_shape = tensor_shape.as_shape([batch] + output_dim)
471 state_shape = [batch] + tensor_shape.as_shape(flat_state).as_list()
472 return tensor_shape.as_shape(state_shape)
530 input_shape = tensor_shape.as_shape(input_shape)
570 InputSpec(shape=[None] + tensor_shape.as_shape(dim).as_list())
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | parsing_ops.py | 269 feature_tensor_shape = tensor_shape.as_shape(feature.shape)
945 dense_shapes = [tensor_shape.as_shape(shape) for shape in dense_shapes]
1445 tensor_shape.as_shape(shape).as_proto()
1449 tensor_shape.as_shape(shape).as_proto()
1781 context_dense_shapes = [tensor_shape.as_shape(shape).as_proto()
1783 feature_list_dense_shapes = [tensor_shape.as_shape(shape).as_proto()
2076 dense_shapes = [tensor_shape.as_shape(shape) for shape in dense_shapes]
|
| D | init_ops_v2_test.py | 46 self.assertEqual(tensor_shape.as_shape(shape), t1.shape)
47 self.assertEqual(tensor_shape.as_shape(shape), t2.shape)
58 self.assertEqual(tensor_shape.as_shape(shape), t1.shape)
59 self.assertEqual(tensor_shape.as_shape(shape), t2.shape)
|
| D | rnn_cell_impl.py | 142 p = tensor_shape.as_shape(prefix)
155 s = tensor_shape.as_shape(suffix)
161 shape = tensor_shape.as_shape(p_static).concatenate(s_static)
|
| D | nn_ops.py | 980 tensor_shape.as_shape(input.shape),
981 tensor_shape.as_shape(filters.shape),
4026 noise_shape_ = tensor_shape.as_shape(noise_shape)
|
| D | variable_scope.py | 677 shape = tensor_shape.as_shape(shape)
852 shape = tensor_shape.as_shape(shape)
|
| /external/tensorflow/tensorflow/python/data/util/ |
| D | convert.py | 53 shape_like = tensor_shape.as_shape(shape_like)
|
| D | structure.py | 432 self._shape = tensor_shape.as_shape(shape)
506 self._dense_shape = tensor_shape.as_shape(dense_shape)
|
| D | structure_test.py | 73 tensor_shape.as_shape(expected).is_compatible_with(actual))
|
| /external/tensorflow/tensorflow/contrib/framework/python/ops/ |
| D | script_ops.py | 137 output_types, tensor_shape.as_shape, output_shapes)
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | iterator_ops.py | 186 output_types, tensor_shape.as_shape, output_shapes)
274 output_types, tensor_shape.as_shape, output_shapes)
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| D | tensor_signature.py | 75 other = tensor_shape.as_shape(other)
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | execute.py | 145 shape = tensor_shape.as_shape(v)
|
| D | backprop.py | 80 return tensor_shape.as_shape(value).as_proto()
82 return [tensor_shape.as_shape(v).as_proto() for v in value]
|
| /external/tensorflow/tensorflow/python/ops/distributions/ |
| D | distribution.py | 671 return tensor_shape.as_shape(self._batch_shape())
705 return tensor_shape.as_shape(self._event_shape())
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | batching.py | 547 output_types, tensor_shape.as_shape, output_shapes)
|
| /external/tensorflow/tensorflow/python/training/ |
| D | input.py | 672 shape_list = [tensor_shape.as_shape(s) for s in shape_list]
|
| /external/tensorflow/tensorflow/python/training/tracking/ |
| D | util.py | 261 shape_object = tensor_shape.as_shape(shape)
|
12