Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.cc197 llvm_ir::PrimitiveTypeToIrType(to_type, module)); in EmitIntegralToFloating()
202 llvm_ir::PrimitiveTypeToIrType(to_type, module)); in EmitIntegralToFloating()
235 llvm_ir::PrimitiveTypeToIrType(PRED, module_)); in EmitIntegerUnaryOp()
239 llvm_ir::PrimitiveTypeToIrType(to_type, module_), in EmitIntegerUnaryOp()
252 auto to_ir_component_type = llvm_ir::PrimitiveTypeToIrType( in EmitIntegerUnaryOp()
278 llvm_ir::PrimitiveTypeToIrType(to_type, module_)); in EmitIntegerUnaryOp()
292 llvm_ir::PrimitiveTypeToIrType(op->shape().element_type(), module_); in EmitIntegerUnaryOp()
309 llvm_ir::PrimitiveTypeToIrType(op->shape().element_type(), module_); in EmitIntegerUnaryOp()
323 llvm_ir::PrimitiveTypeToIrType(PRED, module_)); in EmitIntegerUnaryOp()
354 llvm_ir::PrimitiveTypeToIrType(to_component_type, module_)), in EmitFloatUnaryOp()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dtuple_ops.cc46 llvm::ConstantInt::get(PrimitiveTypeToIrType(PRED, module), 0), in EmitTupleSelect()
77 b->CreatePointerCast(operands[i], PrimitiveTypeToIrType(TUPLE, module)), in EmitTuple()
Dsort_util.cc121 llvm_ir::PrimitiveTypeToIrType(PRED, module), "compare_return_buffer", in EmitCompareLoopBody()
331 llvm_ir::PrimitiveTypeToIrType( in EmitSortInPlace()
Dllvm_util.cc139 llvm::Type* PrimitiveTypeToIrType(PrimitiveType element_type, in PrimitiveTypeToIrType() function
224 llvm::Type* result_type = PrimitiveTypeToIrType(shape.element_type(), module); in ShapeToIrType()
366 return b->CreateZExt(comparison_result, llvm_ir::PrimitiveTypeToIrType( in EmitComparison()
Dllvm_util.h128 llvm::Type* PrimitiveTypeToIrType(PrimitiveType element_type,
Dfused_ir_emitter.cc173 operand_elemental_ir_types.push_back(llvm_ir::PrimitiveTypeToIrType( in HandleTuple()
Dir_array.cc348 PrimitiveTypeToIrType(shape_.element_type(), module) in EmitArrayElementAddress()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Delemental_ir_emitter.cc291 llvm_ir::PrimitiveTypeToIrType(input_type, module_)); in EmitDeviceFunctionCall()
294 llvm_ir::PrimitiveTypeToIrType(output_type, module_), // Return type. in EmitDeviceFunctionCall()
363 llvm_ir::PrimitiveTypeToIrType(operand_element_type, module_), in MakeElementGenerator()
447 b()->CreateAlloca(llvm_ir::PrimitiveTypeToIrType( in MakeElementGenerator()
Dir_emitter.cc678 Alloca(llvm_ir::PrimitiveTypeToIrType( in HandleReduce()
785 llvm_ir::PrimitiveTypeToIrType( in ComputeNestedElement()
Dir_emitter_unnested.cc839 llvm_ir::PrimitiveTypeToIrType(operand_element_type, in HandleSelectAndScatter()
915 llvm_ir::PrimitiveTypeToIrType(PRED, in HandleSelectAndScatter()
927 llvm::ConstantInt::get(llvm_ir::PrimitiveTypeToIrType( in HandleSelectAndScatter()
1182 llvm::Value* input_address = Alloca(llvm_ir::PrimitiveTypeToIrType( in EmitScatter()
2660 llvm::Type* element_type = llvm_ir::PrimitiveTypeToIrType( in EmitPrologueForOneReduction()
3064 llvm_ir::PrimitiveTypeToIrType(param->shape().element_type(), in EmitKernel()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc651 llvm_ir::PrimitiveTypeToIrType(operand_element_type, module_), in EmitElementalReduceWindow()
799 llvm_ir::PrimitiveTypeToIrType(operand_element_type, module_), in HandleSelectAndScatter()
879 llvm::ConstantInt::get(llvm_ir::PrimitiveTypeToIrType(PRED, module_), 0), in HandleSelectAndScatter()
973 llvm_ir::PrimitiveTypeToIrType(lhs_element_type, module_); in EmitElementalConvolution()
1521 llvm_ir::PrimitiveTypeToIrType(element_type, module_); in CreateShardedVectorType()
1825 llvm_ir::PrimitiveTypeToIrType(accumulator_type, module_), "accumulator", in EmitElementalReduce()
2358 llvm::ConstantInt::get(llvm_ir::PrimitiveTypeToIrType(PRED, module_), 0)); in HandleWhile()
2488 llvm_ir::PrimitiveTypeToIrType(primitive_type, module_)); in EmitTransferElements()
2563 llvm_ir::PrimitiveTypeToIrType(PRED, module_), 0), in HandleConditional()
3063 llvm_ir::PrimitiveTypeToIrType(return_type, module_), in EmitThreadLocalCall()
Dvector_support_library.cc33 scalar_type_ = llvm_ir::PrimitiveTypeToIrType( in VectorSupportLibrary()