Home
last modified time | relevance | path

Searched refs:InputsShapePtr (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/ps/
Doptimizer_info_builder.h37 … const Values &values, const Lengths &lens, const InputsShapePtr &inputs_shape, size_t worker_num,
41 … const Lengths &lens, const InputsShapePtr &inputs_shape, size_t worker_num,
50 const Lengths &lens, const InputsShapePtr &inputs_shape = nullptr);
60 const InputsShapePtr &inputs_shape, size_t worker_num,
69 const InputsShapePtr &inputs_shape, size_t worker_num,
78 const InputsShapePtr &inputs_shape, size_t worker_num,
Doptimizer_info_builder.cc28 … const Lengths &lens, const InputsShapePtr &inputs_shape, size_t worker_num, in Build()
58 const InputsShapePtr &inputs_shape) { in GenInputAddrPtr()
115 … const Lengths &lens, const InputsShapePtr &, size_t, in BuildInputs() argument
147 … const Lengths &lens, const InputsShapePtr &inputs_shape, size_t, in BuildInputs()
206 … const Lengths &lens, const InputsShapePtr &inputs_shape, size_t, in BuildInputs()
Dparameter_server.h160 std::unordered_map<Key, InputsShapePtr> optim_inputs_shape_;
161 std::unordered_map<Key, InputsShapePtr> original_optim_inputs_shape_;
Dconstants.h208 using InputsShapePtr = std::shared_ptr<std::vector<std::shared_ptr<std::vector<size_t>>>>; variable
Dparameter_server.cc93 InputsShapePtr inputs_shape = std::make_shared<InputsShape>(); in InitOptimInputsShape()
95 InputsShapePtr original_inputs_shape = std::make_shared<InputsShape>(); in InitOptimInputsShape()