Skip to content

[TF] Helpers to facilitate the move to swift-apis. #25101

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 5 commits into from
May 28, 2019
Merged

[TF] Helpers to facilitate the move to swift-apis. #25101

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
1df089d
Exposed some properties to help facilitate the move to swift-apis.
eaplatanios May 28, 2019
71e77ef
Minor bug fix.
eaplatanios May 28, 2019
2169a5a
Tweak
rxwei May 28, 2019
397416e
Bug fix.
eaplatanios May 28, 2019
b432e94
Merge branch 'move-helpers' of github.com:eaplatanios/swift-language …
eaplatanios May 28, 2019
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions lib/SIL/SILFunctionBuilder.cpp
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -88,6 +88,8 @@ void SILFunctionBuilder::addFunctionAttributes(SILFunction *F,
vjpName = SILDeclRef(vjpFn).mangle();
// Get lowered argument indices.
auto paramIndices = A->getParameterIndices();
if (!paramIndices)
continue;
auto loweredParamIndices = paramIndices->getLowered(
F->getASTContext(),
decl->getInterfaceType()->castTo<AnyFunctionType>());
Expand Down
15 changes: 8 additions & 7 deletions stdlib/public/TensorFlow/TensorHandle.swift
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -143,9 +143,8 @@ internal extension TensorHandle {
}
}

internal extension ShapedArray where Scalar : _TensorFlowDataTypeCompatible {
@usableFromInline
@inline(never)
public extension ShapedArray where Scalar : _TensorFlowDataTypeCompatible {
@inlinable
init(cTensorHandle: CTensorHandle) {
internalConsistencyCheck(TFE_TensorHandleIsConcrete(cTensorHandle) != 0)
let status = TF_NewStatus()
Expand All @@ -162,10 +161,11 @@ internal extension ShapedArray where Scalar : _TensorFlowDataTypeCompatible {
/// `ResourceHandle` is the type used by ops to represent TensorFlow "resource"
/// values.
public struct ResourceHandle {
@usableFromInline
let handle: _AnyTensorHandle

@usableFromInline
var _cTensorHandle: CTensorHandle { handle._cTensorHandle }
@inlinable
public var _cTensorHandle: CTensorHandle { handle._cTensorHandle }

@usableFromInline
init(owning cTensorHandle: CTensorHandle) {
Expand All @@ -176,10 +176,11 @@ public struct ResourceHandle {
/// `VariantHandle` is the type used by ops to represent TensorFlow "variant"
/// values.
public struct VariantHandle {
@usableFromInline
let handle: _AnyTensorHandle

@usableFromInline
var _cTensorHandle: CTensorHandle { handle._cTensorHandle }
@inlinable
public var _cTensorHandle: CTensorHandle { handle._cTensorHandle }

@usableFromInline
init(owning cTensorHandle: CTensorHandle) {
Expand Down