Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
hehl2
Torchaudio
Commits
422edb18
"macapp/src/index.ts" did not exist on "0768b1b9079ea75c22b453e1856e1f2841ff8571"
Unverified
Commit
422edb18
authored
Jan 07, 2021
by
Vincent QB
Committed by
GitHub
Jan 07, 2021
Browse files
update data pointer (#1160)
parent
f2da5861
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
8 additions
and
8 deletions
+8
-8
torchaudio/csrc/transducer.cpp
torchaudio/csrc/transducer.cpp
+8
-8
No files found.
torchaudio/csrc/transducer.cpp
View file @
422edb18
...
...
@@ -43,10 +43,10 @@ int64_t cpu_rnnt_loss(torch::Tensor acts,
std
::
vector
<
float
>
cpu_workspace
(
cpu_size_bytes
/
sizeof
(
float
),
0
);
compute_rnnt_loss
(
acts
.
data
<
float
>
(),
grads
.
data
<
float
>
(),
labels
.
data
<
int
>
(),
label_lengths
.
data
<
int
>
(),
input_lengths
.
data
<
int
>
(),
alphabet_size
,
minibatch_size
,
costs
.
data
<
float
>
(),
compute_rnnt_loss
(
acts
.
data
_ptr
<
float
>
(),
grads
.
data
_ptr
<
float
>
(),
labels
.
data
_ptr
<
int
>
(),
label_lengths
.
data
_ptr
<
int
>
(),
input_lengths
.
data
_ptr
<
int
>
(),
alphabet_size
,
minibatch_size
,
costs
.
data
_ptr
<
float
>
(),
cpu_workspace
.
data
(),
options
);
return
0
;
...
...
@@ -59,10 +59,10 @@ int64_t cpu_rnnt_loss(torch::Tensor acts,
std
::
vector
<
double
>
cpu_workspace
(
cpu_size_bytes
/
sizeof
(
double
),
0
);
compute_rnnt_loss_fp64
(
acts
.
data
<
double
>
(),
grads
.
data
<
double
>
(),
labels
.
data
<
int
>
(),
label_lengths
.
data
<
int
>
(),
input_lengths
.
data
<
int
>
(),
alphabet_size
,
minibatch_size
,
costs
.
data
<
double
>
(),
compute_rnnt_loss_fp64
(
acts
.
data
_ptr
<
double
>
(),
grads
.
data
_ptr
<
double
>
(),
labels
.
data
_ptr
<
int
>
(),
label_lengths
.
data
_ptr
<
int
>
(),
input_lengths
.
data
_ptr
<
int
>
(),
alphabet_size
,
minibatch_size
,
costs
.
data
_ptr
<
double
>
(),
cpu_workspace
.
data
(),
options
);
return
0
;
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment