Searched refs:init_tensors (Results 1 – 4 of 4) sorted by relevance
236 init_tensors = []241 init_tensors.append(tensor)242 base_graph = base_graph or init_tensors[0].graph251 for init_tensor in init_tensors:264 ops_to_visit = [_as_operation(t) for t in init_tensors
151 bool init_tensors();
560 bool IoMemMgr::init_tensors() { in init_tensors() function in example::IoMemMgr649 io_mem_mgr_.init_tensors(), in mem_alloc()
298 init_tensors = variable_utils.convert_variables_to_tensors(init)299 nest.assert_same_structure(init_tensors, entry, expand_composites=True)