Home
last modified time | relevance | path

Searched refs:root_is_integral (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc1423 bool root_is_integral = ShapeUtil::ElementIsIntegral(root_shape); in MatchReductionGenerator() local
1450 return [root_is_integral](llvm::IRBuilder<>* b, llvm::Value* lhs, in MatchReductionGenerator()
1452 return root_is_integral ? b->CreateAdd(lhs, rhs) in MatchReductionGenerator()
1457 return [root_is_integral](llvm::IRBuilder<>* b, llvm::Value* lhs, in MatchReductionGenerator()
1459 return root_is_integral ? b->CreateMul(lhs, rhs) in MatchReductionGenerator()