Home
last modified time | relevance | path

Searched defs:parameter_shape (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore-src/source/tests/ut/python/parallel/
Dtest_using_seed_for_initializer.py21 parameter_shape = [16, 4] variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/hal/hardware/
Dgpu_inference_session.cc153 auto parameter_shape = AnfAlgo::GetOutputDeviceShape(parameter, 0); in CompareInput() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ir/
Dparam_info.h100 const std::vector<int64_t> &parameter_shape() const { return parameter_shape_; } in parameter_shape() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/
Dparameter_manager.cc411 auto parameter_shape = GetNodeShape(parameter); in HandleNoUsedParameter() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/pi/graph_guard/
Dguard_utils.cc932 auto parameter_shape = p->parameter_shape(); in ToStringAttr() local