Home
last modified time | relevance | path

Searched defs:ref_shape (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore-src/source/tests/ut/python/ops/
Dtest_ops.py627 def __init__(self, ref_shape, dtype=np.float32, use_locking=False): argument
666 def __init__(self, ref_shape, dtype=np.float32, use_locking=False): argument
679 def __init__(self, ref_shape, dtype=np.float32): argument
692 def __init__(self, ref_shape, dtype=np.float32): argument
705 def __init__(self, ref_shape, dtype=np.float32): argument
718 def __init__(self, ref_shape, axis, dtype=np.float32): argument
731 def __init__(self, ref_shape, dtype=np.float32): argument
744 def __init__(self, ref_shape, dtype=np.float32): argument
757 def __init__(self, ref_shape, dtype=np.float32, use_locking=False): argument
770 def __init__(self, ref_shape, dtype=np.float32, use_locking=False): argument
[all …]
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/cxx_api/llm_engine/
Dllm_engine_impl.cc298 auto ref_shape = in GetModelInfo() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/ascend_ge/
Dge_graph_executor.cc285 void GeGraphExecutor::SetRefShape(std::vector<int64_t> *ref_shape, bool dyn, std::string tensor_nam… in SetRefShape()
392 auto &ref_shape = ref_data_infos_.front().shape; in CheckRefDataInfo() local
431 auto &ref_shape = ref_data_infos_.front().shape; in InitMaxShapeParam() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/
Dslice_prepose_pass.cc318 …posePass::SliceParamDeBroadcast(const CNodePtr &slice_cnode, const std::vector<int64_t> &ref_shape, in SliceParamDeBroadcast()
415 const std::vector<int64_t> &ref_shape) { in SiblingsAreSameSlice()