Home
last modified time | relevance | path

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

/third_party/mindspore/tests/st/ops/cpu/
Dtest_rl_buffer_net.py28 def create_tensor(capcity, shapes, types): function
39 self.buffer = create_tensor(capcity, shapes, types)
/third_party/mindspore/tests/st/ops/gpu/
Dtest_rl_buffer_net.py28 def create_tensor(capcity, shapes, dtypes): function
38 self.buffer = create_tensor(capcity, shapes, types)