Home
last modified time | relevance | path

Searched defs:grad_tensor (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32_grad/
Dsoftmax_crossentropy_fp32_tests.cc65 lite::Tensor grad_tensor(TypeId::kNumberTypeFloat32, dim_y); in TEST_F() local
/third_party/mindspore/mindspore/ccsrc/runtime/framework/
Dgraph_compiler.cc506 …piler::AddGradAddrToBucket(const GraphId &graph_id, const std::vector<tensor::TensorPtr> &grad_ten… in AddGradAddrToBucket()
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dsession_basic.cc2511 …Basic::AddGradAddrToBucket(const GraphId &graph_id, const std::vector<tensor::TensorPtr> &grad_ten… in AddGradAddrToBucket()