Searched refs:test_tensor_shape (Results 1 – 5 of 5) sorted by relevance
70 const TensorShape test_tensor_shape = {10, 200}; in TEST() local71 Tensor test_tensor(DT_FLOAT, test_tensor_shape); in TEST()132 const TensorShape test_tensor_shape = {100, 200}; in TEST() local133 Tensor test_tensor(DT_FLOAT, test_tensor_shape); in TEST()
402 def test_tensor_shape(op_name: str) -> str: function412 tensor_size_ex = test_tensor_shape(op_name)
3739 test_tensor_shape = [None, 1, 1, 1]3742 tensor_spec.TensorSpec(shape=test_tensor_shape, dtype=dtypes.float32)3752 for d in test_tensor_shape
4574 def test_tensor_shape(self): member in TestScript
5827 FileName: ./python/pyarmnn/test/test_tensor_shape.py