| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_sharding.py | 179 shape = tensor_shape.as_shape(shape)
190 return tensor_shape.as_shape(dims)
223 shape = tensor_shape.as_shape(shape)
259 shape = tensor_shape.as_shape(shape)
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | python_api_parameter_converter_test.py | 132 ("ShapeFromShape", "shape", tensor_shape.as_shape([1, 2]),
133 tensor_shape.as_shape([1, 2])),
134 ("ShapeFromInt", "shape", 1, tensor_shape.as_shape(1)),
135 ("ShapeFromNone", "shape", None, tensor_shape.as_shape(None)),
136 ("ShapeFromList", "shape", [1, 2, 3], tensor_shape.as_shape([1, 2, 3])),
144 tensor_shape.as_shape(1),
145 tensor_shape.as_shape(None),
146 tensor_shape.as_shape([2, 3])
184 self.assertEqual(params, [[1, 2, 3], tensor_shape.as_shape([1, 2]), 10.0])
|
| D | tensor_shape.py | 965 other = as_shape(other)
1006 other = as_shape(other)
1022 other = as_shape(other)
1136 other = as_shape(other)
1180 other = as_shape(other)
1273 other = as_shape(other)
1281 other = as_shape(other)
1297 def as_shape(shape): function
|
| D | indexed_slices.py | 230 self._shape = tensor_shape.as_shape(shape)
237 self._indices_shape = tensor_shape.as_shape(indices_shape).with_rank(1)
|
| D | op_def_util.cc | 155 static PyObject* as_shape = GetRegisteredPyObject("tf.as_shape"); in operator ()() local
160 result.reset(PyObject_CallFunctionObjArgs(as_shape, value, nullptr)); in operator ()()
|
| D | tensor_shape_test.py | 412 tensor_shape.as_shape(proto))
418 tensor_shape.as_shape(proto))
420 partial_proto_shape = tensor_shape.as_shape(
|
| D | constant_op.py | 304 shape = tensor_shape.as_shape(shape)
|
| D | type_spec_test.py | 60 self.x_shape = tensor_shape.as_shape(x_shape)
62 self.y_shape = tensor_shape.as_shape(y_shape)
|
| D | sparse_tensor.py | 311 self._shape = tensor_shape.as_shape(shape)
|
| D | op_def_library.py | 221 return tensor_shape.as_shape(v).as_proto()
827 _pywrap_utils.RegisterPyObject("tf.as_shape", tensor_shape.as_shape)
|
| /external/tensorflow/tensorflow/python/data/util/ |
| D | convert.py | 53 shape_like = tensor_shape.as_shape(shape_like)
|
| /external/tensorflow/tensorflow/python/ops/structured/ |
| D | structured_tensor_slice_test.py | 135 a_shape = tensor_shape.as_shape(a.shape)
136 b_shape = tensor_shape.as_shape(b.shape)
|
| D | structured_tensor.py | 191 shape = tensor_shape.as_shape(shape)
1047 shape = tensor_shape.as_shape(shape)
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | input_layer_test.py | 68 def as_shape(shape): function
83 self.x_shape = as_shape(x_shape)
85 self.y_shape = as_shape(y_shape)
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | parameter_server_strategy_v2.py | 592 shape = tensor_shape.as_shape(shape)
659 shape=tensor_shape.as_shape(partition_shape),
|
| /external/tensorflow/tensorflow/python/ops/ |
| 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)
55 self.assertEqual(tensor_shape.as_shape(shape), t1.shape)
56 self.assertEqual(tensor_shape.as_shape(shape), t2.shape)
|
| D | parsing_config.py | 420 self.dense_shapes = [tensor_shape.as_shape(s) for s in dense_shapes]
568 feature_tensor_shape = tensor_shape.as_shape(feature.shape)
578 self.dense_shapes.append(tensor_shape.as_shape(feature.shape))
590 self.dense_shapes.append(tensor_shape.as_shape(feature.shape))
|
| D | tensor_array_ops.py | 140 self._element_shape = [tensor_shape.as_shape(element_shape)]
460 self._element_shape = [tensor_shape.as_shape(element_shape)]
707 self._element_shape = tensor_shape.as_shape(element_shape)
1334 self._element_shape = tensor_shape.as_shape(element_shape)
|
| D | nn_ops.py | 1159 tensor_shape.as_shape(input.shape),
1160 tensor_shape.as_shape(filters.shape),
1215 filter_shape = tensor_shape.as_shape(filter_shape)
1216 input_shape = tensor_shape.as_shape(input_shape)
4990 noise_shape_ = tensor_shape.as_shape(noise_shape)
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_one_hot_op_test.py | 173 indices_shape = tensor_shape.as_shape(indices_shape)
|
| D | ragged_tensor.py | 1458 shape = tensor_shape.as_shape(shape)
1487 flat_shape = tensor_shape.as_shape([None] + shape[self.ragged_rank + 1:])
1760 shape = tensor_shape.as_shape(shape)
2299 self._shape = tensor_shape.as_shape(shape)
2858 shape = tensor_shape.as_shape(shape)
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | iterator_ops.py | 199 tensor_shape.as_shape,
273 tensor_shape.as_shape,
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | execute.py | 211 shape = tensor_shape.as_shape(v)
|
| D | backprop.py | 95 return tensor_shape.as_shape(value).as_proto()
97 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())
|