Searched defs:ref_shape (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/tests/ut/python/ops/ |
| D | test_ops.py | 416 def __init__(self, ref_shape, dtype=np.float32, use_locking=False): argument
455 def __init__(self, ref_shape, dtype=np.float32, use_locking=False): argument
468 def __init__(self, ref_shape, dtype=np.float32): argument
481 def __init__(self, ref_shape, dtype=np.float32): argument
494 def __init__(self, ref_shape, dtype=np.float32): argument
507 def __init__(self, ref_shape, dtype=np.float32, use_locking=False): argument
520 def __init__(self, ref_shape, dtype=np.float32, use_locking=False): argument
533 def __init__(self, ref_shape, dtype=np.float32, use_locking=False): argument
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/graph/ |
| D | slice_prepose_pass.cc | 309 …posePass::SliceParamDeBroadcast(const CNodePtr &slice_cnode, const std::vector<int64_t> &ref_shape, in SliceParamDeBroadcast()
404 const std::vector<int64_t> &ref_shape) { in SiblingsAreSameSlice()
|