Searched refs:ComputeOffsetPointer (Results 1 – 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | vector_support_library.h | 148 llvm::Value* ComputeOffsetPointer(llvm::Value* base_pointer, 150 llvm::Value* ComputeOffsetPointer(llvm::Value* base_pointer, in ComputeOffsetPointer() function 152 return ComputeOffsetPointer( in ComputeOffsetPointer() 155 llvm::Value* ComputeOffsetPointer(llvm::Value* base_pointer, in ComputeOffsetPointer() function 157 return ComputeOffsetPointer(base_pointer, b()->getInt64(offset_elements)); in ComputeOffsetPointer() 164 return LoadVector(ComputeOffsetPointer(base_pointer, offset_elements)); in LoadVector() 175 return LoadScalar(ComputeOffsetPointer(base_pointer, offset_elements)); in LoadScalar() 186 StoreVector(value, ComputeOffsetPointer(base_pointer, offset_elements)); in StoreVector() 197 StoreScalar(value, ComputeOffsetPointer(base_pointer, offset_elements)); in StoreScalar() 208 return LoadBroadcast(ComputeOffsetPointer(base_pointer, offset_elements)); in LoadBroadcast()
|
D | tiled_dot_emitter.cc | 45 pointers_.push_back(vsl_->ComputeOffsetPointer(matrix, total_offset)); in MemoryTile() 266 llvm::Value* base_pointer = vsl_.ComputeOffsetPointer(rhs_, offset); in LoadRhsTile() 365 vsl_.ComputeOffsetPointer(lhs_, total_offset); in EmitInnerLoopEpilogue() 592 vsl_.ComputeOffsetPointer(lhs_, total_offset); in EmitInnerLoopEpilogue()
|
D | vector_support_library.cc | 201 llvm::Value* VectorSupportLibrary::ComputeOffsetPointer( in ComputeOffsetPointer() function in xla::cpu::VectorSupportLibrary
|