Searched defs:grad_tensor (Results 1 – 3 of 3) sorted by relevance
65 lite::Tensor grad_tensor(TypeId::kNumberTypeFloat32, dim_y); in TEST_F() local
506 …piler::AddGradAddrToBucket(const GraphId &graph_id, const std::vector<tensor::TensorPtr> &grad_ten… in AddGradAddrToBucket()
2511 …Basic::AddGradAddrToBucket(const GraphId &graph_id, const std::vector<tensor::TensorPtr> &grad_ten… in AddGradAddrToBucket()