Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.cc231 llvm_ir::PrimitiveTypeToIrType(to_type, module)); in EmitIntegralToFloating()
236 llvm_ir::PrimitiveTypeToIrType(to_type, module)); in EmitIntegralToFloating()
269 llvm_ir::PrimitiveTypeToIrType(PRED, module_)); in EmitIntegerUnaryOp()
273 llvm_ir::PrimitiveTypeToIrType(to_type, module_), in EmitIntegerUnaryOp()
286 auto to_ir_component_type = llvm_ir::PrimitiveTypeToIrType( in EmitIntegerUnaryOp()
312 llvm_ir::PrimitiveTypeToIrType(to_type, module_)); in EmitIntegerUnaryOp()
326 llvm_ir::PrimitiveTypeToIrType(op->shape().element_type(), module_); in EmitIntegerUnaryOp()
343 llvm_ir::PrimitiveTypeToIrType(op->shape().element_type(), module_); in EmitIntegerUnaryOp()
357 llvm_ir::PrimitiveTypeToIrType(PRED, module_)); in EmitIntegerUnaryOp()
401 llvm_ir::PrimitiveTypeToIrType(to_component_type, module_)), in EmitFloatUnaryOp()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dtuple_ops.cc43 b->CreatePointerCast(operands[i], PrimitiveTypeToIrType(TUPLE, module)); in EmitTuple()
78 llvm_ir::PrimitiveTypeToIrType(element_shape.element_type(), module); in EmitTupleAllocasAtFunctionEntry()
Dsort_util.cc128 llvm::Type* pred_type = llvm_ir::PrimitiveTypeToIrType(PRED, module); in EmitCompareLoopBody()
358 llvm_ir::PrimitiveTypeToIrType( in EmitSortInPlace()
Dfused_ir_emitter.cc128 element_ir_types.push_back(llvm_ir::PrimitiveTypeToIrType( in HandleTuple()
Dllvm_util.cc146 llvm::Type* PrimitiveTypeToIrType(PrimitiveType element_type, in PrimitiveTypeToIrType() function
233 llvm::Type* result_type = PrimitiveTypeToIrType(shape.element_type(), module); in ShapeToIrType()
369 return b->CreateZExt(comparison_result, llvm_ir::PrimitiveTypeToIrType( in EmitComparison()
Dllvm_util.h124 llvm::Type* PrimitiveTypeToIrType(PrimitiveType element_type,
Dir_array.cc487 llvm::Type* type = PrimitiveTypeToIrType(shape_.element_type(), module); in EmitArrayElementAddress()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dtarget_util.cc203 llvm_ir::PrimitiveTypeToIrType(input_type, module)); in EmitDeviceFunctionCall()
206 llvm_ir::PrimitiveTypeToIrType(output_type, module), // Return type. in EmitDeviceFunctionCall()
Dir_emitter.cc195 Load(llvm_ir::PrimitiveTypeToIrType(element_type, module_), in MaybeEmitDirectAtomicOperation()
Dir_emitter_unnested.cc1982 llvm_ir::PrimitiveTypeToIrType(operand_shape.element_type(), module_), in EmitSelectAndScatter()
2074 llvm_ir::PrimitiveTypeToIrType(PRED, module_), "select_return_buffer", in EmitSelectAndScatter()
2093 llvm_ir::PrimitiveTypeToIrType(PRED, module_), 0), in EmitSelectAndScatter()
2415 llvm_ir::PrimitiveTypeToIrType(desc.updates_shape.element_type(), in EmitScatter()
3596 llvm_ir::PrimitiveTypeToIrType(result_shape.element_type(), module_); in GenerateReductionCodegenState()
4235 llvm_ir::PrimitiveTypeToIrType(param_shape.element_type(), module_), in EmitHlo021Tile()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc741 llvm_ir::PrimitiveTypeToIrType(operand_element_type, module_), in HandleSelectAndScatter()
828 llvm::ConstantInt::get(llvm_ir::PrimitiveTypeToIrType(PRED, module_), 0), in HandleSelectAndScatter()
1582 llvm_ir::PrimitiveTypeToIrType(element_type, module_); in CreateShardedVectorType()
2579 llvm::ConstantInt::get(llvm_ir::PrimitiveTypeToIrType(PRED, module_), 0)); in HandleWhile()
2765 llvm_ir::PrimitiveTypeToIrType(primitive_type, module_); in EmitTransferElements()
2845 llvm_ir::PrimitiveTypeToIrType(PRED, module_), 0), in HandleConditional()
Dvector_support_library.cc33 scalar_type_ = llvm_ir::PrimitiveTypeToIrType( in VectorSupportLibrary()
Ddot_op_emitter.cc842 float_type = llvm_ir::PrimitiveTypeToIrType(C64, module); in EmitCallToRuntime()
848 float_type = llvm_ir::PrimitiveTypeToIrType(C128, module); in EmitCallToRuntime()