projects / platform / upstream / tensorflow.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 538cab2) | patch
[XLA] Don't call MultidimensionalIndexToLinearIndex in HloEvaluator's convolution...
authorJustin Lebar <jlebar@google.com>
Fri, 6 Apr 2018 02:30:10 +0000 (19:30 -0700)
committerTensorFlower Gardener <gardener@tensorflow.org>
Fri, 6 Apr 2018 02:32:55 +0000 (19:32 -0700)
commit05f665543adecddfbfc44f7bcd4e74d983c2c442
tree0b69ef834c5fe6ceefa2c24f95a79ceb1c3ed3fctree | snapshot
parent538cab2870fd02b4d89e0534ad52573e67f16606commit | diff
[XLA] Don't call MultidimensionalIndexToLinearIndex in HloEvaluator's convolution routine.

Before: ConvolutionTest/0.StridedFilter (41812 ms)
After:  ConvolutionTest/0.StridedFilter (28054 ms)

Speedup: 42 / 28 = 1.5x
PiperOrigin-RevId: 191835735
tensorflow/compiler/xla/service/hlo_evaluator.cc diff | blob | history
Domain: Machine Learning / ML Framework;