Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/grad/
Dvariable.cc46 …ValuePtrList flatten_values = PyNativeAlgo::DataConvert::FlattenTensorSeqInValueSeq(gradient_value… in PostProcess() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/graph_compiler/
Dbackend_base.cc86 std::vector<tensor::TensorPtr> flatten_values; in PushTensor() local
92 size_t index, std::map<size_t, std::vector<tensor::TensorPtr>> *flatten_values, in PushTupleTensor()
150 std::map<size_t, std::vector<tensor::TensorPtr>> flatten_values; in GetRunGraphInputs() local
Dbackend.cc258 std::vector<tensor::TensorPtr> flatten_values; in MsRunGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/pi/auto_grad/
Dfunction_node.cc194 ValuePtrList flatten_values = ValuePtrList(edges_.size(), GetGrad()[0]); in ApplyNative() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/grad/function/
Dfunc_grad.cc707 auto flatten_values = PyNativeAlgo::DataConvert::FlattenTensorSeqInValue(sens); in OnsLike() local