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

[XLA] In LiteralUtil::StridedConfig: choose the larger dimension between the

Browse Source 
source and destination shapes' minor-most dimension index.

PiperOrigin-RevId: 163732014
This commit is contained in:
Kay Zhu 2017-07-31 12:36:40 -07:00 committed by TensorFlower Gardener
parent f9c644693e
commit 5887cc10e7
1 changed files with 4 additions and 4 deletions

8
tensorflow/compiler/xla/literal_util.cc
View File

@ -43,10 +43,10 @@ Literal::StrideConfig::StrideConfig(
base(dimensions.size(), 0),
step(dimensions.size(), 1) {
if (!dimensions.empty()) {
// Selects the shape with the highest minor dimension as the one upon
// where to run the tight stride loop.
if (source_shape.layout().minor_to_major()[0] >=
dest_shape.layout().minor_to_major()[0]) {
// Selects the shape with the largest minor dimension as the one upon
// which to run the tight stride loop.
if (dimensions[source_shape.layout().minor_to_major()[0]] >=
dimensions[dest_shape.layout().minor_to_major()[0]]) {
minor_dimension = source_shape.layout().minor_to_major()[0];
dest_stride = IndexUtil::GetDimensionStride(dest_shape, minor_dimension);
} else {