Home
last modified time | relevance | path

Searched refs:tangents_out (Results 1 – 20 of 20) sorted by relevance

/dports/math/py-jax/jax-0.2.9/jax/interpreters/
H A Dad.py262 results = (_, tangents_out) = yield args, kwargs
263 yield results, [type(r) is not Zero for r in tangents_out]
352 primals_out, tangents_out = split_list(outs, [len(outs) // 2])
353 tangents_out = map(recast_to_float0, primals_out, tangents_out)
354 return map(partial(JVPTracer, self), primals_out, tangents_out)
366 tangents_out = custom_lin_p.bind(
369 tangents_out = map(recast_to_float0, primals_out, tangents_out)
472 tangents_out = list(tangents_out)
651 primals_out, tangents_out = yield (primals, tangents), {}
652 out_nonzeros = [type(t) is not Zero for t in tangents_out]
[all …]
H A Dinvertible_ad.py148 _, tangents_out = ad.jvp(lu.wrap_init(fun)).call_wrapped(primals, tangents)
149 return primals_out, tangents_out
/dports/math/py-jax/jax-0.2.9/jax/
H A Dcustom_derivatives.py245 tangents_out, out_tree2 = tree_flatten(py_tangents_out)
251 tangent_avals_out = [raise_to_shaped(core.get_aval(t), weak_type=False) for t in tangents_out]
265 yield primals_out + tangents_out, out_tree
326 primals_out, tangents_out = split_list(outs, [len(outs) // 2])
327 tangents_out = map(ad.recast_to_float0, primals_out, tangents_out)
328 return primals_out, tangents_out
620 tangents_out = ad.custom_lin_p.bind(
622 tangents_out = map(ad.recast_to_float0, primals_out, tangents_out)
623 return primals_out, tangents_out
H A Dapi.py1815 tangents_out = eval_jaxpr(jaxpr, consts, *tangents)
1817 out_pvals, tangents_out))
2592 tangents_out = [rule(t, ans, *primals) for rule, t in zip(jvprules, tangents)
2594 return ans, functools.reduce(ad.add_tangents, tangents_out, ad_util.Zero.from_value(ans))
/dports/games/doomsday/doomsday-2.3.1/doomsday/external/assimp/code/
H A DFBXMeshGeometry.h145 void ReadVertexDataTangents( std::vector<aiVector3D>& tangents_out, const Scope& source,
H A DFBXMeshGeometry.cpp558 void MeshGeometry::ReadVertexDataTangents(std::vector<aiVector3D>& tangents_out, const Scope& sourc… in ReadVertexDataTangents() argument
564 ResolveVertexDataArray(tangents_out,source,MappingInformationType,ReferenceInformationType, in ReadVertexDataTangents()
/dports/graphics/qt5-3d/kde-qt3d-5.15.2p39/src/3rdparty/assimp/code/
H A DFBXMeshGeometry.h146 void ReadVertexDataTangents( std::vector<aiVector3D>& tangents_out, const Scope& source,
H A DFBXMeshGeometry.cpp558 void MeshGeometry::ReadVertexDataTangents(std::vector<aiVector3D>& tangents_out, const Scope& sourc… in ReadVertexDataTangents() argument
564 ResolveVertexDataArray(tangents_out,source,MappingInformationType,ReferenceInformationType, in ReadVertexDataTangents()
/dports/graphics/urho3d/Urho3D-1.7.1/Source/ThirdParty/Assimp/code/
H A DFBXMeshGeometry.cpp465 void MeshGeometry::ReadVertexDataTangents(std::vector<aiVector3D>& tangents_out, const Scope& sourc… in ReadVertexDataTangents() argument
470 ResolveVertexDataArray(tangents_out,source,MappingInformationType,ReferenceInformationType, in ReadVertexDataTangents()
H A DFBXDocument.h860 void ReadVertexDataTangents(std::vector<aiVector3D>& tangents_out, const Scope& source,
/dports/devel/upp/upp/bazaar/plugin/assimp/code/FBX/
H A DFBXMeshGeometry.h150 void ReadVertexDataTangents( std::vector<aiVector3D>& tangents_out, const Scope& source,
H A DFBXMeshGeometry.cpp572 void MeshGeometry::ReadVertexDataTangents(std::vector<aiVector3D>& tangents_out, const Scope& sourc… in ReadVertexDataTangents() argument
578 ResolveVertexDataArray(tangents_out,source,MappingInformationType,ReferenceInformationType, in ReadVertexDataTangents()
/dports/multimedia/assimp/assimp-5.1.3/code/AssetLib/FBX/
H A DFBXMeshGeometry.h150 void ReadVertexDataTangents( std::vector<aiVector3D>& tangents_out, const Scope& source,
H A DFBXMeshGeometry.cpp591 void MeshGeometry::ReadVertexDataTangents(std::vector<aiVector3D>& tangents_out, const Scope& sourc… in ReadVertexDataTangents() argument
597 ResolveVertexDataArray(tangents_out,source,MappingInformationType,ReferenceInformationType, in ReadVertexDataTangents()
/dports/devel/godot/godot-3.2.3-stable/thirdparty/assimp/code/FBX/
H A DFBXMeshGeometry.h150 void ReadVertexDataTangents( std::vector<aiVector3D>& tangents_out, const Scope& source,
H A DFBXMeshGeometry.cpp572 void MeshGeometry::ReadVertexDataTangents(std::vector<aiVector3D>& tangents_out, const Scope& sourc… in ReadVertexDataTangents() argument
578 ResolveVertexDataArray(tangents_out,source,MappingInformationType,ReferenceInformationType, in ReadVertexDataTangents()
/dports/devel/godot-tools/godot-3.2.3-stable/thirdparty/assimp/code/FBX/
H A DFBXMeshGeometry.h150 void ReadVertexDataTangents( std::vector<aiVector3D>& tangents_out, const Scope& source,
H A DFBXMeshGeometry.cpp572 void MeshGeometry::ReadVertexDataTangents(std::vector<aiVector3D>& tangents_out, const Scope& sourc… in ReadVertexDataTangents() argument
578 ResolveVertexDataArray(tangents_out,source,MappingInformationType,ReferenceInformationType, in ReadVertexDataTangents()
/dports/math/py-jax/jax-0.2.9/jax/_src/lax/
H A Dcontrol_flow.py1525 tangents_out = [next(tangents_out_iter) if nz else ad_util.Zero.from_value(p)
1527 return primals_out, tangents_out
H A Dlax.py5767 tangents_out = tuple(t if type(t) is ad_util.Zero else t[idx] for t in tangents)
5768 return tuple(primals[:-1]), tangents_out