Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dtuple_ops.cc46 llvm::ConstantInt::get(PrimitiveTypeToIrType(PRED, module), 0), in EmitTupleSelect()
66 b->CreatePointerCast(operands[i], PrimitiveTypeToIrType(TUPLE, module)); in EmitTuple()
100 llvm_ir::PrimitiveTypeToIrType(element_shape.element_type(), module); in EmitTupleAllocasAtFunctionEntry()
Dsort_util.cc122 llvm_ir::PrimitiveTypeToIrType(PRED, module), "compare_return_buffer", in EmitCompareLoopBody()
333 llvm_ir::PrimitiveTypeToIrType( in EmitSortInPlace()
Dfused_ir_emitter.cc127 operand_elemental_ir_types.push_back(llvm_ir::PrimitiveTypeToIrType( in HandleTuple()
Dllvm_util.cc145 llvm::Type* PrimitiveTypeToIrType(PrimitiveType element_type, in PrimitiveTypeToIrType() function
232 llvm::Type* result_type = PrimitiveTypeToIrType(shape.element_type(), module); in ShapeToIrType()
368 return b->CreateZExt(comparison_result, llvm_ir::PrimitiveTypeToIrType( in EmitComparison()
Dllvm_util.h131 llvm::Type* PrimitiveTypeToIrType(PrimitiveType element_type,
Dir_array.cc458 PrimitiveTypeToIrType(shape_.element_type(), module) in EmitArrayElementAddress()
/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.cc225 llvm_ir::PrimitiveTypeToIrType(to_type, module)); in EmitIntegralToFloating()
230 llvm_ir::PrimitiveTypeToIrType(to_type, module)); in EmitIntegralToFloating()
263 llvm_ir::PrimitiveTypeToIrType(PRED, module_)); in EmitIntegerUnaryOp()
267 llvm_ir::PrimitiveTypeToIrType(to_type, module_), in EmitIntegerUnaryOp()
280 auto to_ir_component_type = llvm_ir::PrimitiveTypeToIrType( in EmitIntegerUnaryOp()
306 llvm_ir::PrimitiveTypeToIrType(to_type, module_)); in EmitIntegerUnaryOp()
320 llvm_ir::PrimitiveTypeToIrType(op->shape().element_type(), module_); in EmitIntegerUnaryOp()
337 llvm_ir::PrimitiveTypeToIrType(op->shape().element_type(), module_); in EmitIntegerUnaryOp()
351 llvm_ir::PrimitiveTypeToIrType(PRED, module_)); in EmitIntegerUnaryOp()
395 llvm_ir::PrimitiveTypeToIrType(to_component_type, module_)), in EmitFloatUnaryOp()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dtarget_util.cc202 llvm_ir::PrimitiveTypeToIrType(input_type, module)); in EmitDeviceFunctionCall()
205 llvm_ir::PrimitiveTypeToIrType(output_type, module), // Return type. in EmitDeviceFunctionCall()
Dir_emitter_unnested.cc2270 llvm_ir::PrimitiveTypeToIrType(operand_shape.element_type(), in EmitSelectAndScatterFromMlir()
2359 llvm_ir::PrimitiveTypeToIrType(PRED, in EmitSelectAndScatterFromMlir()
2377 llvm::ConstantInt::get(llvm_ir::PrimitiveTypeToIrType( in EmitSelectAndScatterFromMlir()
2702 llvm_ir::PrimitiveTypeToIrType(desc.updates_shape.element_type(), in EmitScatter()
4247 llvm::Type* element_type = llvm_ir::PrimitiveTypeToIrType( in EmitPrologueForReduction()
4288 llvm::Type* primitive_type = llvm_ir::PrimitiveTypeToIrType( in EmitPrologueForReduction()
4867 llvm_ir::PrimitiveTypeToIrType(param_shape.element_type(), module_), in EmitHlo021Tile()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc718 llvm_ir::PrimitiveTypeToIrType(operand_element_type, module_), in HandleSelectAndScatter()
798 llvm::ConstantInt::get(llvm_ir::PrimitiveTypeToIrType(PRED, module_), 0), in HandleSelectAndScatter()
1486 llvm_ir::PrimitiveTypeToIrType(element_type, module_); in CreateShardedVectorType()
2419 llvm::ConstantInt::get(llvm_ir::PrimitiveTypeToIrType(PRED, module_), 0)); in HandleWhile()
2589 llvm_ir::PrimitiveTypeToIrType(primitive_type, module_)); in EmitTransferElements()
2665 llvm_ir::PrimitiveTypeToIrType(PRED, module_), 0), in HandleConditional()
Ddot_op_emitter.cc775 float_type = llvm_ir::PrimitiveTypeToIrType(C64, module); in EmitCallToRuntime()
781 float_type = llvm_ir::PrimitiveTypeToIrType(C128, module); in EmitCallToRuntime()
Dvector_support_library.cc33 scalar_type_ = llvm_ir::PrimitiveTypeToIrType( in VectorSupportLibrary()