Searched refs:RLBufferSample (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_rl_buffer_op.py | 54 class RLBufferSample(nn.Cell): class
56 super(RLBufferSample, self).__init__()
90 buffer_sample = RLBufferSample(capcity=5, batch_size=3, shapes=[(4,), (2,), (1,), (4,)], types=[
|
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_rl_buffer_op.py | 54 class RLBufferSample(nn.Cell): class
56 super(RLBufferSample, self).__init__()
91 buffer_sample = RLBufferSample(capcity=5, batch_size=3, shapes=[(4,), (2,), (1,), (4,)], types=[
|