Home
last modified time | relevance | path

Searched defs:blas (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dplugin_registry.h107 std::map<PluginId, BlasFactory> blas; member
117 PluginId blas, dnn, fft, rng; member
Dplugin.cc33 PluginConfig& PluginConfig::SetBlas(PluginId blas) { in SetBlas()
Dplugin.h76 PluginId blas() const { return blas_; } in blas() function
Dblas.cc21 namespace blas { namespace
Dstream.cc1452 if (blas::BlasSupport *blas = stream->parent_->AsBlas()) { in Run() local
/external/eigen/blas/
DCMakeLists.txt15 add_custom_target(blas) target
/external/tensorflow/tensorflow/core/kernels/
Dgpu_utils.cc157 if (auto* blas = stream_exec->AsBlas()) { in LogConvAutotuneResults() local
204 if (auto* blas = stream_exec->AsBlas()) { in LogFusedConvForwardAutotuneResults() local
/external/deqp/external/vulkancts/modules/vulkan/ray_tracing/
DvktRayTracingCallableShadersTests.cpp740 for (auto& blas : bottomLevelAccelerationStructures) in runTest() local
DvktRayTracingTraversalControlTests.cpp662 for (auto& blas : bottomLevelAccelerationStructures) in runTest() local
DvktRayTracingShaderBindingTableTests.cpp926 for (auto& blas : bottomLevelAccelerationStructures) in runTest() local
DvktRayTracingAccelerationStructuresTests.cpp1561 for (auto& blas : bottomLevelAccelerationStructures) in runTest() local
2223 …de::SharedPtr<BottomLevelAccelerationStructure> blas = de::SharedPtr<BottomLevelAccelerationStr… in iterate() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_algorithm_picker.cc441 if (auto* blas = stream_exec_->AsBlas()) { in PickBestAlgorithmNoCacheCuda() local
/external/deqp/external/vulkancts/modules/vulkan/ray_query/
DvktRayQueryTraversalControlTests.cpp1042 for (auto& blas : bottomLevelAccelerationStructures) in fillCommandBuffer() local
1947 for (auto& blas : rayQueryBottomLevelAccelerationStructures) in iterate() local
DvktRayQueryAccelerationStructuresTests.cpp1062 for (auto& blas : bottomLevelAccelerationStructures) in fillCommandBuffer() local
2482 for (auto& blas : bottomLevelAccelerationStructures) in runTest() local
3173 …de::SharedPtr<BottomLevelAccelerationStructure> blas = de::SharedPtr<BottomLevelAccelerationStr… in iterate() local
DvktRayQueryCullRayFlagsTests.cpp1065 for (auto& blas : bottomLevelAccelerationStructures) in fillCommandBuffer() local
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_blas.cc2705 gpu::ROCMBlas *blas = new gpu::ROCMBlas(rocm_executor); in initialize_rocblas() local
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_blas.cc3771 gpu::CUDABlas *blas = new gpu::CUDABlas(cuda_executor); in initialize_cublas() local