Home
last modified time | relevance | path

Searched defs:ir_builder (Results 1 – 23 of 23) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dloop_emitter.cc36 llvm::IRBuilder<>* ir_builder) in LoopEmitter()
41 llvm::IRBuilder<>* ir_builder) in LoopEmitter()
56 const std::vector<IrArray>& target_arrays, llvm::IRBuilder<>* ir_builder) { in MakeBodyEmitterForMultiOutputFusion()
77 llvm::IRBuilder<>* ir_builder) in LoopEmitter()
Dops.cc43 tensorflow::StringPiece name, llvm::IRBuilder<>* ir_builder) { in EmitDynamicUpdateSliceInPlaceImpl()
89 llvm::IRBuilder<>* ir_builder) { in EmitDynamicUpdateSliceInPlace()
120 llvm::IRBuilder<>* ir_builder) { in EmitFusedDynamicUpdateSliceInPlaceImpl()
163 llvm::IRBuilder<>* ir_builder) { in EmitFusedDynamicUpdateSliceInPlace()
174 llvm::IRBuilder<>* ir_builder) { in EmitParallelFusedDynamicUpdateSliceInPlace()
Dtuple_ops.cc34 llvm::Value* on_false, llvm::IRBuilder<>* ir_builder, in EmitTupleSelect()
75 llvm::IRBuilder<>* ir_builder, llvm::Module* module) { in EmitTuple()
89 llvm::IRBuilder<>* ir_builder, in EmitGetTupleElement()
Dllvm_util.cc49 llvm::Module* ModuleFromIRBuilder(llvm::IRBuilder<>* ir_builder) { in ModuleFromIRBuilder()
88 llvm::IRBuilder<>* ir_builder) { in EmitCallToIntrinsic()
104 llvm::IRBuilder<>* ir_builder) { in EmitFloatMax()
115 llvm::IRBuilder<>* ir_builder) { in EmitFloatMin()
126 llvm::IRBuilder<>* ir_builder) { in EmitBufferIndexingGEP()
144 llvm::IRBuilder<>* ir_builder) { in EmitBufferIndexingGEP()
234 const Shape& shape, int32* shape_size, llvm::IRBuilder<>* ir_builder) { in EncodeSelfDescribingShapeConstant()
379 llvm::IRBuilder<>* ir_builder, in EmitAllocaAtFunctionEntry()
387 llvm::IRBuilder<>* ir_builder, int alignment) { in EmitAllocaAtFunctionEntryWithCount()
403 llvm::IRBuilder<>* ir_builder) { in CreateBasicBlock()
[all …]
Dir_array.cc33 llvm::IRBuilder<>* ir_builder) in Index()
80 const Shape& shape, llvm::IRBuilder<>* ir_builder) in Index()
224 const IrArray::Index& index, llvm::IRBuilder<>* ir_builder, in EmitArrayElementAddress()
282 llvm::IRBuilder<>* ir_builder, in EmitReadArrayElement()
310 llvm::IRBuilder<>* ir_builder) { in BumpIndex()
Dllvm_loop.cc50 llvm::Value* end_index, llvm::Value* step, llvm::IRBuilder<>* ir_builder, in EmitForLoop()
59 void ForLoop::Emit(llvm::IRBuilder<>* ir_builder) { in Emit()
148 llvm::IRBuilder<>* ir_builder) { in GetLoopMetadata()
173 llvm::IRBuilder<>* ir_builder) { in CreateLoopBB()
Dkernel_support_library.h38 : ir_builder_(ir_builder), in ir_builder_() argument
150 llvm::IRBuilder<>* ir_builder, tensorflow::StringPiece kernel_name, in EmitAndCallOutlinedKernel()
163 llvm::IRBuilder<>* ir_builder, tensorflow::StringPiece kernel_name, in EmitAndCallOutlinedKernel()
Dllvm_loop.h173 explicit ForLoopNest(llvm::IRBuilder<>* ir_builder) in ForLoopNest()
176 ForLoopNest(tensorflow::StringPiece name, llvm::IRBuilder<>* ir_builder) in ForLoopNest()
Dkernel_support_library.cc69 llvm::IRBuilder<>* ir_builder, tensorflow::StringPiece kernel_name, in EmitAndCallOutlinedKernel()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dparallel_loop_emitter.cc35 const LaunchDimensions& launch_dimensions, llvm::IRBuilder<>* ir_builder) in ParallelLoopEmitter()
42 const LaunchDimensions& launch_dimensions, llvm::IRBuilder<>* ir_builder) in ParallelLoopEmitter()
49 const LaunchDimensions& launch_dimensions, llvm::IRBuilder<>* ir_builder) in ParallelLoopEmitter()
Dhlo_to_ir_bindings.h39 llvm::IRBuilder<>* ir_builder, llvm::Module* llvm_module, in HloToIrBindings()
Delemental_ir_emitter.cc64 llvm::IRBuilder<>* ir_builder, NestedComputer compute_nested) in GpuElementalIrEmitter()
/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.h37 llvm::Module* module, llvm::IRBuilder<>* ir_builder) in ElementalIrEmitter()
57 llvm::IRBuilder<>* ir_builder() const { return ir_builder_; } in ir_builder() function
Delemental_ir_emitter.cc57 llvm::IRBuilder<>* ir_builder) { in EmitReducePrecisionFloat()
164 llvm::IRBuilder<>* ir_builder) { in EmitF32ToBF16()
177 llvm::IRBuilder<>* ir_builder) { in EmitBF16ToF32()
188 llvm::IRBuilder<>* ir_builder) { in EmitIntegralToFloating()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_function.cc52 llvm::IRBuilder<>* ir_builder, in IrFunction()
198 llvm::IRBuilder<>* ir_builder, tensorflow::StringPiece name, in GetArrayFunctionCallArguments()
234 llvm::IRBuilder<>* ir_builder, llvm::Function* parallel_function, in EmitCallToParallelForkJoin()
Dparallel_loop_emitter.cc28 const DynamicLoopBounds* dynamic_loop_bounds, llvm::IRBuilder<>* ir_builder) in ParallelLoopEmitter()
Dllvm_ir_runtime.cc55 llvm::IRBuilder<> ir_builder(vector_tanh_body); in EmitVectorF32TanhIfNeeded() local
116 llvm::IRBuilder<> ir_builder(vector_exp_body); in EmitVectorF32ExpIfNeeded() local
199 llvm::IRBuilder<> ir_builder(vector_log_body); in EmitVectorF32LogIfNeeded() local
Ddot_op_emitter.cc51 TileLoader(VectorSupportLibrary* vsl, llvm::IRBuilder<>* ir_builder, in TileLoader()
149 llvm::IRBuilder<>* ir_builder) in ColumnMajorMatrixVectorProductEmitter()
373 llvm::IRBuilder<>* ir_builder) in RowMajorMatrixVectorProductEmitter()
529 llvm::Value* executable_run_options_value, llvm::IRBuilder<>* ir_builder, in DotOpEmitter()
548 llvm::Value* executable_run_options_value, llvm::IRBuilder<>* ir_builder, in EmitDotOperation()
Dir_emitter.h105 llvm::IRBuilder<>* ir_builder() { return &ir_builder_; } in ir_builder() function
Dir_emitter.cc1380 return [root_is_integral](llvm::IRBuilder<>* ir_builder, llvm::Value* lhs, in MatchReductionGenerator()
1387 return [](llvm::IRBuilder<>* ir_builder, llvm::Value* lhs, in MatchReductionGenerator()
1391 return [](llvm::IRBuilder<>* ir_builder, llvm::Value* lhs, in MatchReductionGenerator()
1396 llvm::IRBuilder<>* ir_builder, llvm::Value* lhs, in MatchReductionGenerator()
1413 llvm::IRBuilder<>* ir_builder, llvm::Value* lhs, in MatchReductionGenerator()
1414 llvm::Value* rhs) { in MatchReductionGenerator()
2590 llvm::IRBuilder<>* ir_builder, llvm::Value* prof_counter, in UpdateProfileCounter()
2601 llvm::IRBuilder<>* ir_builder) { in ReadCycleCounter()
2629 void IrEmitter::ProfilingState::RecordCycleStart(llvm::IRBuilder<>* ir_builder, in RecordCycleStart()
2639 void IrEmitter::ProfilingState::RecordCycleDelta(llvm::IRBuilder<>* ir_builder, in RecordCycleDelta()
[all …]
Dvector_support_library.cc26 llvm::IRBuilder<>* ir_builder, in VectorSupportLibrary()
411 LlvmVariable::LlvmVariable(llvm::Type* type, llvm::IRBuilder<>* ir_builder) in LlvmVariable()
Dvector_support_library.h216 llvm::IRBuilder<>* ir_builder() const { return ir_builder_; } in ir_builder() function
/external/mesa3d/src/compiler/glsl/
Dir_builder.cpp29 namespace ir_builder { namespace