experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

use smaller slice length for 32-bit arch

Browse Source 
Signed-off-by: Saurabh Deoras <sdeoras@gmail.com>
This commit is contained in:
Saurabh Deoras 2019-01-10 14:49:47 -08:00
parent 1271cb8142
commit 12c9bb638e
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/go/attrs.go
View File

@ -170,7 +170,8 @@ func listAttribute(op *Operation, cname *C.char, meta C.TF_AttrMetadata) (interf
} }
// A []C.int64_t slice backed by C memory. // A []C.int64_t slice backed by C memory.
// See: https://github.com/golang/go/wiki/cgo#turning-c-arrays-into-go-slices // See: https://github.com/golang/go/wiki/cgo#turning-c-arrays-into-go-slices
slice := (*[1 << 30]C.int64_t)(unsafe.Pointer(dim))[:numDim:numDim] // Using [1<<27] instead of [1<<30] so it works on 32-bit architecture
slice := (*[1 << 27]C.int64_t)(unsafe.Pointer(dim))[:numDim:numDim]
list[i] = makeCShape(slice) list[i] = makeCShape(slice)
} }
return list, nil return list, nil