| /external/tensorflow/tensorflow/compiler/xla/service/ |
| elemental_ir_emitter.h | 38 : ir_builder_(ir_builder),
57 llvm::IRBuilder<>* ir_builder() const { return ir_builder_; }
142 return ir_builder_->getIntN(128, 0);
145 llvm::IRBuilder<>* const ir_builder_; member in class:xla::ElementalIrEmitter::llvm
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| kernel_support_library.h | 38 : ir_builder_(ir_builder),
59 For(name, /*start=*/ir_builder_->getInt64(start),
60 /*end=*/ir_builder_->getInt64(end),
61 /*step=*/ir_builder_->getInt64(step), for_body_generator);
90 /*step=*/ir_builder_->getInt64(step), peel_first_iteration,
106 For(name, /*start=*/ir_builder_->getInt64(start),
107 /*end=*/ir_builder_->getInt64(end),
108 /*step=*/ir_builder_->getInt64(step), for_body_generator);
176 llvm::IRBuilder<>* ir_builder_; member in class:xla::KernelSupportLibrary
|
| loop_emitter.h | 87 llvm::IRBuilder<>* ir_builder_; member in class:xla::llvm_ir::LoopEmitter
|
| fused_ir_emitter.h | 60 ir_builder_(elemental_emitter->ir_builder()),
99 llvm::IRBuilder<>* ir_builder_; member in class:xla::FusedIrEmitter
|
| llvm_loop.h | 181 ir_builder_(ir_builder) {}
260 llvm::IRBuilder<>* ir_builder_; member in class:xla::llvm_ir::ForLoopNest
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| hlo_to_ir_bindings.h | 43 ir_builder_(ir_builder),
46 &ir_builder_->getContext()) {}
107 llvm::IRBuilder<>* ir_builder_; member in class:xla::gpu::HloToIrBindings
|
| ir_emitter.h | 163 llvm::IRBuilder<> ir_builder_; member in class:xla::gpu::IrEmitter
|
| elemental_ir_emitter.cc | 89 converted_operands[i] = ir_builder_->CreateFPCast(
90 converted_operands[i], ir_builder_->getFloatTy());
109 result = ir_builder_->CreateFPCast(result, ir_builder_->getHalfTy());
212 return ir_builder_->CreateFDiv(llvm::ConstantFP::get(llvm_ty, 1), sqrt);
289 ir_builder_->GetInsertBlock()->getModule()->getOrInsertFunction(
296 return ir_builder_->CreateCall(callee, llvm_ir::AsArrayRef(operands));
300 llvm::Value* block_id = ir_builder_->CreateIntCast(
302 {}, {}, ir_builder_),
303 ir_builder_->getIntNTy(128), /*isSigned=*/true, "block.id")
367 llvm_ir::ForLoopNest loops(IrName(hlo), ir_builder_); local
426 llvm_ir::ForLoopNest loops(IrName(hlo), ir_builder_); member in namespace:xla::gpu
[all...] |
| ir_emitter.cc | 59 ir_builder_(module_->getContext()),
61 &ir_emitter_context->buffer_assignment(), &ir_builder_, module_,
64 ir_builder_.setFastMathFlags(llvm_ir::GetFastMathFlags(
74 .EmitReadArrayElement(index, &ir_builder_);
78 *hlo, GpuElementalIrEmitter(hlo_module_config_, module_, &ir_builder_,
121 /*alignment=*/1, GetBasePointer(*operand), &ir_builder_, module_));
151 llvm_ir::EmitTuple(GetIrArray(*tuple, *tuple), base_ptrs, &ir_builder_,
173 ir_builder_.CreateCall(emitted_function, arguments);
193 llvm::Value* source = ir_builder_.CreateLoad(source_address, "source");
200 {output_address->getType()}, &ir_builder_);
504 llvm_ir::ForLoopNest loop_nest(IrName(dot), &ir_builder_); local
649 llvm_ir::ForLoopNest loops(IrName(reduce, "inner"), &ir_builder_); local
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| dot_op_emitter.h | 146 llvm::IRBuilder<>* ir_builder_; member in class:xla::cpu::DotOpEmitter
|
| ir_function.h | 100 llvm::IRBuilder<>* ir_builder_; member in class:xla::cpu::IrFunction
|
| ir_emitter.h | 105 llvm::IRBuilder<>* ir_builder() { return &ir_builder_; }
420 llvm::IRBuilder<> ir_builder_; member in class:xla::cpu::IrEmitter
|
| vector_support_library.h | 216 llvm::IRBuilder<>* ir_builder() const { return ir_builder_; }
273 llvm::IRBuilder<>* ir_builder_; member in class:xla::cpu::VectorSupportLibrary
292 llvm::IRBuilder<>* ir_builder_; member in class:xla::cpu::LlvmVariable
|
| dot_op_emitter.cc | 159 ir_builder_(ir_builder),
160 ksl_(ir_builder_),
161 vsl_(scalar_type_, /*vector_size=*/tile_rows_, ir_builder_, "") {
172 return TileLoader(&vsl_, ir_builder_, /*matrix=*/lhs_,
206 llvm::IRBuilder<>* ir_builder_; member in class:xla::cpu::__anon38877::ColumnMajorMatrixVectorProductEmitter
235 EmitOuterLoopBody(ir_builder_->getInt64(column_limit), column_remainder,
267 llvm::Value* columns_llvm = ir_builder_->getInt64(columns);
278 /*end=*/ir_builder_->CreateAdd(columns_llvm, current_tile_col),
283 ir_builder_->CreateMul(col, ir_builder_->getInt64(m_))
416 llvm::IRBuilder<>* ir_builder_; member in class:xla::cpu::__anon38877::RowMajorMatrixVectorProductEmitter
764 llvm_ir::ForLoopNest loop_nest(llvm_ir::IrName(&dot_), ir_builder_); local
[all...] |
| ir_emitter.cc | 91 ir_builder_(llvm_module->getContext()),
101 ir_builder_.setFastMathFlags(llvm_ir::GetFastMathFlags(
151 module_, &ir_builder_, num_dynamic_loop_bounds_));
159 ir_builder_.CreateBitCast(GetEmittedValueFor(bitcast->operand(0)),
309 GetEmittedValueFor(operand), &ir_builder_, module_);
323 GetEmittedValueFor(on_false), &ir_builder_, module_); local
368 llvm_ir::EmitTuple(infeed_array, tuple_element_addresses, &ir_builder_,
392 shape, &shape_length, &ir_builder_));
397 llvm::Type* int32_type = ir_builder_.getInt32Ty();
417 ir_builder_.getVoidTy()
491 llvm_ir::EmitTuple(GetIrArrayFor(tuple), base_ptrs, &ir_builder_, module_); local
559 &ir_builder_); local
666 llvm_ir::ForLoopNest source_loops(IrName(select_and_scatter), &ir_builder_); local
688 &ir_builder_); local
1020 llvm_ir::ForLoopNest loops(IrName(convolution, "inner"), &ir_builder_); local
[all...] |
| /frameworks/compile/mclinker/tools/mcld/ |
| Main.cpp | 91 ir_builder_(module_, config_) {
112 mcld::IRBuilder ir_builder_; member in class:__anon46048::Driver
987 if (!InitializeInputs(ir_builder_, actions)) {
[all...] |
