HomeSort by relevance Sort by last modified time
    Searched refs:VectorSupportLibrary (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
vector_support_library.cc 25 VectorSupportLibrary::VectorSupportLibrary(PrimitiveType primitive_type,
47 void VectorSupportLibrary::AssertCorrectTypes(
59 llvm::Value* VectorSupportLibrary::Mul(llvm::Value* lhs, llvm::Value* rhs) {
64 llvm::Value* VectorSupportLibrary::MulInternal(llvm::Value* lhs,
73 llvm::Value* VectorSupportLibrary::Add(llvm::Value* lhs, llvm::Value* rhs) {
78 llvm::Value* VectorSupportLibrary::Sub(llvm::Value* lhs, llvm::Value* rhs) {
83 llvm::Value* VectorSupportLibrary::Max(llvm::Value* lhs, llvm::Value* rhs) {
92 llvm::Value* VectorSupportLibrary::Floor(llvm::Value* a) {
98 llvm::Value* VectorSupportLibrary::Div(llvm::Value* lhs, llvm::Value* rhs)
    [all...]
vector_support_library.h 42 class VectorSupportLibrary {
44 // This VectorSupportLibrary instance remembers `primitive_type` and
48 VectorSupportLibrary(PrimitiveType primitive_type, int64 vector_size,
113 // makes kernel IR generation code written using VectorSupportLibrary (its
305 VectorVariable(VectorSupportLibrary* vector_support,
314 ScalarVariable(VectorSupportLibrary* vector_support,
326 TileVariable(VectorSupportLibrary* vector_support,
tiled_dot_emitter.cc 36 MemoryTile(VectorSupportLibrary* vsl, llvm::IRBuilder<>* b,
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_,
795 VectorSupportLibrary vsl(scalar_type(), 1, b_, "gemm");
803 void TiledSmallGemmEmitter::HandleResiduesOnK(VectorSupportLibrary* vsl
    [all...]
llvm_ir_runtime.cc 120 VectorSupportLibrary vsl(F32, vector_width, b, "exp_f32");
158 // VectorSupportLibrary (intentionally) can't juggle more than one type at a
178 VectorSupportLibrary vsl(F32, vector_width, b, "log_f32");
213 // VectorSupportLibrary (intentionally) can't juggle more than one type at a

Completed in 112 milliseconds