Home
last modified time | relevance | path

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

/third_party/mindspore/tests/st/ops/cpu/
Dtest_rl_buffer_net.py36 class RLBuffer(nn.Cell): class
38 super(RLBuffer, self).__init__()
75 buffer = RLBuffer(batch_size=32, capcity=100, shapes=[(4,), (2,), (1,), (4,)], types=[
/third_party/mindspore/tests/st/ops/gpu/
Dtest_rl_buffer_net.py35 class RLBuffer(nn.Cell): class
37 super(RLBuffer, self).__init__()
75 buffer = RLBuffer(batch_size=32, capcity=100, shapes=[(4,), (2,), (1,), (4,)], types=[