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

Merge pull request #2 from tatatodd/xla_ir_array_fix

Browse Source 
[XLA] Explicitly instantiate Permute template func
This commit is contained in:
Todd Wang 2017-06-20 14:16:50 -07:00 committed by GitHub
commit 49292d15d2
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(dimension_mapping, multidim()); Permute<std::vector, llvm::Value*>(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);