Lines Matching refs:shape_a

112         shape_a = (N, L)
115         shape_a = (L, N)
118         shape_a = (N, L)
121         shape_a = (L, N)
126     A = te.placeholder(shape_a, name="A", dtype=dtype)
310         shape_a = (N, L)
313         shape_a = (L, N)
316         shape_a = (N, L)
319         shape_a = (L, N)
328         a_np = np.random.uniform(size=shape_a).astype(np.float16)
340         a_np = np.random.randint(low=-128, high=127, size=shape_a).astype(np.int8)
352         a_np_int = np.random.randint(low=-8, high=7, size=shape_a).astype(np.int32)
371         a_np_int = np.random.randint(low=0, high=1, size=shape_a).astype(np.int32)