Home
last modified time | relevance | path

Searched defs:reshape_with_axis (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/tests/st/ops/gpu/
Dtest_time_distributed_op.py28 def __init__(self, cell, time_axis, reshape_with_axis=None): argument
/third_party/mindspore/tests/st/ops/cpu/
Dtest_time_distributed_op.py28 def __init__(self, cell, time_axis, reshape_with_axis=None): argument
/third_party/mindspore/mindspore/nn/layer/
Dtimedistributed.py104 def __init__(self, layer, time_axis, reshape_with_axis=None): argument