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

Update ir_array.cc

Browse Source
This commit is contained in:
Todd Wang 2017-06-21 17:54:15 -07:00 committed by GitHub
parent 70960eb13a
commit 03da611346
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/compiler/xla/service/llvm_ir/ir_array.cc
View File

@ -158,7 +158,7 @@ IrArray::Index IrArray::Index::SourceIndexOfTranspose(
tensorflow::gtl::ArraySlice<int64> dimension_mapping, tensorflow::gtl::ArraySlice<int64> dimension_mapping,
llvm::IRBuilder<>* builder) const { llvm::IRBuilder<>* builder) const {
std::vector<llvm::Value*> operand_multidim_index = std::vector<llvm::Value*> operand_multidim_index =
Permute<std::vector, llvm::Value*>(dimension_mapping, multidim()); Permute(dimension_mapping, multidim());
if (linear() != nullptr && if (linear() != nullptr &&
ShapeUtil::TransposeIsBitcast(operand_shape, shape, dimension_mapping)) { ShapeUtil::TransposeIsBitcast(operand_shape, shape, dimension_mapping)) {
return Index(operand_multidim_index, linear(), operand_shape); return Index(operand_multidim_index, linear(), operand_shape);