Home
last modified time | relevance | path

Searched defs:backprop_id (Results 1 – 25 of 26) sorted by relevance

12

/dports/science/py-chainer/chainer-7.8.0/chainerx_cc/chainerx/
H A Dbackprop_mode.cc35 for (const BackpropId& backprop_id : backprop_ids) { local
41 for (const BackpropId& backprop_id : backprop_ids) { local
60 BackpropId backprop_id = context.default_backprop_id(); in IsBackpropRequired() local
64 bool IsBackpropRequired(const BackpropId& backprop_id) { in IsBackpropRequired()
H A Darray_body.h73 bool IsGradRequired(const BackpropId& backprop_id) const { in IsGradRequired()
81 static void RequireGrad(const std::shared_ptr<ArrayBody>& body, const BackpropId& backprop_id) { in RequireGrad()
99 const std::shared_ptr<ArrayNode>& GetArrayNode(const BackpropId& backprop_id) const { in GetArrayNode()
108 …bool HasArrayNode(const BackpropId& backprop_id) const { return GetNodeIndex(backprop_id).has_valu… in HasArrayNode()
126 const absl::optional<Array>* GetGrad(const BackpropId& backprop_id) const { in GetGrad()
132 absl::optional<Array>* GetGrad(const BackpropId& backprop_id) { in GetGrad()
H A Dbackward_builder.cc52 const BackpropId& backprop_id = input_array_node->backprop_id(); in CreateInputArrayNodesMap() local
71 const BackpropId& backprop_id = pair.first; in CreateInputArrayNodesMap() local
89 const BackpropId& backprop_id = pair.first; in Define() local
137 std::shared_ptr<OpNode>& BackwardBuilder::FindOrCreateOpNode(const BackpropId& backprop_id) { in FindOrCreateOpNode()
256 for (const BackpropId& backprop_id : retained_graphs) { in AddEdgesFromOpNodeToArrayNodeOfOuterGraphsForRetention() local
270 const BackpropId& backprop_id = tup.first; in AddEdgesFromOpNodeToArrayNodeOfOuterGraphsForRetention() local
H A Darray_body.cc86 …ArrayBody::CreateArrayNode(const std::shared_ptr<ArrayBody>& body, const BackpropId& backprop_id) { in CreateArrayNode()
125 void ArrayBody::SetGrad(Array grad, const BackpropId& backprop_id) { in SetGrad()
131 void ArrayBody::ClearGrad(const BackpropId& backprop_id) { in ClearGrad()
138 ReturnType ArrayBody::GetGradImpl(ThisPtr this_ptr, const BackpropId& backprop_id) { in GetGradImpl()
H A Dcontext.cc129 void Context::ReleaseBackpropId(const BackpropId& backprop_id) { in ReleaseBackpropId()
139 void Context::ReleaseBackpropIdNoExcept(const BackpropId& backprop_id) noexcept { in ReleaseBackpropIdNoExcept()
201 std::string Context::GetBackpropName(const BackpropId& backprop_id) { in GetBackpropName()
208 void Context::CheckBackpropAllowed(const BackpropId& backprop_id) { in CheckBackpropAllowed()
223 void Context::SetBackpropDone(const BackpropId& backprop_id) { in SetBackpropDone()
245 std::vector<BackpropId> Context::GetInnerBackpropIds(const BackpropId& backprop_id) { in GetInnerBackpropIds()
H A Darray_repr_test.cc38 for (const BackpropId& backprop_id : backprop_ids) { local
243 BackpropId backprop_id = backprop_scope.backprop_id(); in TEST() local
263 BackpropId backprop_id = backprop_scope.backprop_id(); in TEST() local
317 BackpropId backprop_id = backprop_scope.backprop_id(); in TEST() local
H A Dbackprop_mode.h21 BackpropMode(const BackpropId& backprop_id, bool backprop) in BackpropMode()
26 const absl::optional<BackpropId>& backprop_id() const { return backprop_id_; } in backprop_id() function
H A Darray_node.h22 ArrayNode(Shape shape, Dtype dtype, Device& device, BackpropId backprop_id) in ArrayNode()
57 const BackpropId& backprop_id() const { return backprop_id_; } in backprop_id() function
H A Dcheck_backward.cc46 const BackpropId& backprop_id, in BackwardGradients()
112 const BackpropId& backprop_id) { in CheckDoubleBackpropOption()
170 const absl::optional<BackpropId>& backprop_id) { in CheckBackwardComputation()
267 const absl::optional<BackpropId>& backprop_id) { in CheckBackward()
342 const absl::optional<BackpropId>& backprop_id) { in CheckDoubleBackwardComputationImpl()
444 const absl::optional<BackpropId>& backprop_id) { in CheckDoubleBackwardComputation()
H A Dbackprop_scope.h23 BackpropId backprop_id() const { return backprop_id_; } in backprop_id() function
H A Dgraph.cc10 std::ostream& operator<<(std::ostream& os, const BackpropId& backprop_id) { return os << backprop_i… in operator <<()
H A Dbackward.cc102 const BackpropId& backprop_id) { in CreateSubgraph()
201 const BackpropId& backprop_id, in BackwardImpl()
241 const BackpropId& backprop_id, in BackwardImpl()
609 const absl::optional<BackpropId>& backprop_id, in Backward()
619 const absl::optional<BackpropId>& backprop_id, in Backward()
632 const absl::optional<BackpropId>& backprop_id, in Grad()
H A Dop_node.cc44 …std::string name, BackpropId backprop_id, size_t input_count, const std::vector<ConstArrayRef>& ou… in CreateWithOutputArrayNodes()
69 OpNode::OpNode(std::string name, BackpropId backprop_id, size_t input_array_node_count) in OpNode()
93 const BackpropId& backprop_id, in AssertOuterGraphsArrayNodesConsistency()
H A Darray_body_leak_detection.cc49 const BackpropId& backprop_id = array_node->backprop_id(); in IsAllArrayBodiesFreed() local
H A Darray_test.cc223 BackpropId backprop_id = backprop_scope.backprop_id(); in TEST_P() local
265 BackpropId backprop_id = backprop_scope.backprop_id(); in TEST_P() local
293 BackpropId backprop_id = backprop_scope.backprop_id(); in TEST_P() local
445 BackpropId backprop_id = backprop_scope.backprop_id(); in TEST_P() local
684 BackpropId backprop_id = backprop_scope.backprop_id(); in TEST_P() local
741 BackpropId backprop_id = backprop_scope.backprop_id(); in TEST_P() local
848 BackpropId backprop_id = backprop_scope.backprop_id(); in TEST_P() local
1074 BackpropId backprop_id = backprop_scope.backprop_id(); in TEST_P() local
1252 BackpropId backprop_id = backprop_scope.backprop_id(); in TEST() local
1268 BackpropId backprop_id = backprop_scope.backprop_id(); in TEST() local
[all …]
H A Dop_node.h123 BackpropId backprop_id() const { return backprop_id_; } in backprop_id() function
H A Dbackward_test.cc231 BackpropId backprop_id = ctx.default_backprop_id(); in TEST_F() local
635 BackpropId backprop_id = backprop_scope.backprop_id(); in TEST_F() local
721 BackpropId backprop_id = backprop_scope.backprop_id(); in TEST_F() local
739 BackpropId backprop_id = backprop_scope.backprop_id(); in TEST_F() local
1397 BackpropId backprop_id = backprop_scope.backprop_id(); in TEST_P() local
1456 BackpropId backprop_id = backprop_scope.backprop_id(); in TEST_P() local
1528 BackpropId backprop_id = backprop_scope.backprop_id(); in TEST_P() local
1636 BackpropId backprop_id = backprop_scope.backprop_id(); in TEST_P() local
1809 BackpropId backprop_id = backprop_scope.backprop_id(); in TEST_P() local
2908 BackpropId backprop_id = backprop_scope.backprop_id(); in TEST() local
[all …]
H A Darray.cc55 BackpropId GetArrayBackpropId(const Array& array, const absl::optional<BackpropId>& backprop_id) { in GetArrayBackpropId()
547 T& Array::RequireGradImpl(T& array, const absl::optional<BackpropId>& backprop_id) { in RequireGradImpl()
675 const absl::optional<BackpropId>& backprop_id, in DebugDumpComputationalGraph()
H A Dbackprop_mode_test.cc414 BackpropId backprop_id = backprop_scope.backprop_id(); in TEST() local
/dports/science/py-chainer/chainer-7.8.0/chainerx_cc/chainerx/python/
H A Dbackward.cc59 const absl::optional<BackpropId>& backprop_id, in InitChainerxBackward()
74 const absl::optional<BackpropId>& backprop_id, in InitChainerxBackward()
79 absl::optional<float> loss_scale) { in InitChainerxBackward()
H A Dbackprop_mode.cc60 …m.def(function_name, [](const BackpropId& backprop_id) { return PyBackpropModeScope<BackpropModeSc… in InitChainerxBackpropModeScope()
73 …m.def("is_backprop_required", [](const BackpropId& backprop_id) { return IsBackpropRequired(backpr… in InitChainerxBackpropMode()
H A Dcheck_backward.cc53 const absl::optional<BackpropId>& backprop_id) { in InitChainerxCheckBackward()
81 const absl::optional<BackpropId>& backprop_id) { in InitChainerxCheckBackward()
H A Dcontext.cc69 … [](Context& self, const BackpropId& backprop_id) { return self.ReleaseBackpropId(backprop_id); }, in InitChainerxContext()
73 …[](Context& self, const BackpropId& backprop_id) { return self.CheckValidBackpropId(backprop_id); … in InitChainerxContext()
H A Dgraph.cc58 c.def("__repr__", [](const BackpropId& backprop_id) { in InitChainerxGraph()
H A Darray.cc623 [](const ArrayBodyPtr& self, const absl::optional<BackpropId>& backprop_id) { in InitChainerxArraySpecial()
628 …[](const ArrayBodyPtr& self, const absl::optional<BackpropId>& backprop_id) { return Array{self}.I… in InitChainerxArraySpecial()
631 [](const ArrayBodyPtr& self, const absl::optional<BackpropId>& backprop_id) { in InitChainerxArraySpecial()
639 … [](const ArrayBodyPtr& self, const absl::optional<BackpropId>& backprop_id) -> ConstArrayBodyPtr { in InitChainerxArraySpecial()
648 …(const ArrayBodyPtr& self, const ArrayBodyPtr& grad, const absl::optional<BackpropId>& backprop_id… in InitChainerxArraySpecial()
670 [](const ArrayBodyPtr& self, const absl::optional<BackpropId>& backprop_id) { in InitChainerxArraySpecial()
692 …[](const ArrayBodyPtr& self, const absl::optional<BackpropId>& backprop_id) { Array{self}.ClearGra… in InitChainerxArraySpecial()
746 [](const ArrayBodyPtr& self, const absl::optional<BackpropId>& backprop_id) { in InitChainerxArraySpecial()

12