def test_dtype(): fname = os.path.join(this_file_dir, 'runtime_analysis/{}/general/dtype.txt'.format(DIM)) if os.path.exists(fname): os.remove(fname) for lib, call in [ (l, c) for l, c in helpers.calls if c not in [helpers.tf_graph_call, helpers.mx_graph_call] ]: time_lib = LIB_DICT[lib] append_to_file(fname, '{}'.format(lib)) x0 = ivy_gen.array([random.uniform(0, 1) for _ in range(DIM)], f=lib) ivy_gen.dtype(x0, f=lib) ivy_gen_w_time.dtype(x0, f=time_lib) TIMES_DICT.clear() for _ in range(100): log_time(fname, 'tb0') ivy_gen_w_time.dtype(x0, f=time_lib) log_time(fname, 'tb4', time_at_start=True) log_time(fname, 'tt0') ivy_gen.dtype(x0, f=lib) log_time(fname, 'tt1', time_at_start=True) write_times() append_to_file(fname, 'end of analysis')
def dtype(self): """ Return container, with all entries replaced with their data types. :return: New datatype container """ return self.map(lambda x, _: _ivy_gen.dtype(x))
def test_dtype(): for lib, call in helpers.calls: if call is helpers.mx_graph_call: # MXNet symbolic does not support dtype continue assert ivy_gen.dtype(ivy_gen.array([0.], f=lib)) == ivy_gen.array( [0.], f=lib).dtype