Searched refs:VectorSupportLibrary (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | vector_support_library.cc | 25 VectorSupportLibrary::VectorSupportLibrary(PrimitiveType primitive_type, in VectorSupportLibrary() function in xla::cpu::VectorSupportLibrary
47 void VectorSupportLibrary::AssertCorrectTypes( in AssertCorrectTypes()
59 llvm::Value* VectorSupportLibrary::Mul(llvm::Value* lhs, llvm::Value* rhs) { in Mul()
64 llvm::Value* VectorSupportLibrary::MulInternal(llvm::Value* lhs, in MulInternal()
73 llvm::Value* VectorSupportLibrary::Add(llvm::Value* lhs, llvm::Value* rhs) { in Add()
78 llvm::Value* VectorSupportLibrary::Sub(llvm::Value* lhs, llvm::Value* rhs) { in Sub()
83 llvm::Value* VectorSupportLibrary::Max(llvm::Value* lhs, llvm::Value* rhs) { in Max()
92 llvm::Value* VectorSupportLibrary::Floor(llvm::Value* a) { in Floor()
98 llvm::Value* VectorSupportLibrary::Div(llvm::Value* lhs, llvm::Value* rhs) { in Div()
107 llvm::Value* VectorSupportLibrary::Clamp(llvm::Value* a, in Clamp()
[all …]
|
| D | tiled_dot_emitter.cc | 36 MemoryTile(VectorSupportLibrary* vsl, llvm::IRBuilder<>* b, in MemoryTile()
95 VectorSupportLibrary* vsl_;
289 VectorSupportLibrary vsl_;
500 VectorSupportLibrary vsl_;
723 void HandleResiduesOnK(VectorSupportLibrary* vsl, llvm::Value* n_start,
725 void HandleResiduesOnM(VectorSupportLibrary* vsl, int64 tile_size_k,
731 void EmitTiledGemm(VectorSupportLibrary* vsl, int64 tile_size_k,
780 VectorSupportLibrary vsl(scalar_type(), current_vectorization_width, b_, in HandleResiduesOnN()
795 VectorSupportLibrary vsl(scalar_type(), 1, b_, "gemm"); in HandleResiduesOnN()
803 void TiledSmallGemmEmitter::HandleResiduesOnK(VectorSupportLibrary* vsl, in HandleResiduesOnK()
[all …]
|
| D | vector_support_library.h | 42 class VectorSupportLibrary {
48 VectorSupportLibrary(PrimitiveType primitive_type, int64 vector_size,
305 VectorVariable(VectorSupportLibrary* vector_support, in VectorVariable()
314 ScalarVariable(VectorSupportLibrary* vector_support, in ScalarVariable()
326 TileVariable(VectorSupportLibrary* vector_support,
|
| D | llvm_ir_runtime.cc | 120 VectorSupportLibrary vsl(F32, vector_width, b, "exp_f32"); in GenerateVF32Exp()
178 VectorSupportLibrary vsl(F32, vector_width, b, "log_f32"); in GenerateVF32Log()
|