Home
last modified time | relevance | path

Searched refs:tensorShapeEqual (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/
Darray_ops_test.py879 def tensorShapeEqual(self, x, y): member in StridedSliceShapeTest
889 self.tensorShapeEqual(a[3:5], tensor_shape.TensorShape([2, None, 7]))
890 self.tensorShapeEqual(a[3:5, :, 4], tensor_shape.TensorShape([2, None]))
891 self.tensorShapeEqual(a[3:5, 3:4, 4], tensor_shape.TensorShape([2, None]))
892 self.tensorShapeEqual(a[3:5, :, 5:10],
894 self.tensorShapeEqual(a[3:5, :, 50:3],
896 self.tensorShapeEqual(a[3:5, :, array_ops.newaxis, 50:3,],
898 self.tensorShapeEqual(a[1:5:2, :, array_ops.newaxis, 50:3,],
900 self.tensorShapeEqual(a[:5:3, :, array_ops.newaxis, 50:3,],
902 self.tensorShapeEqual(a[:2:3, :, array_ops.newaxis, 50:3,],
[all …]