[XLA] Explicitly instantiate Permute template func
This is an attempted fix for older compilers that can't deduce the template args.
This commit is contained in:
parent
3e2093844c
commit
eb197c45ed
@ -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);
|
||||||
|
Loading…
x
Reference in New Issue
Block a user