Home
last modified time | relevance | path

Searched defs:blas (Results 1 – 20 of 20) 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.h1367 blas::BlasSupport *blas = parent()->AsBlas(); in ThenBlasGemm() local
1450 blas::BlasSupport *blas = parent()->AsBlas(); in ThenBlasGemmWithAlgorithm() local
1489 blas::BlasSupport *blas = parent()->AsBlas(); in ThenBlasGemmStridedBatchedWithAlgorithm() local
1607 blas::BlasSupport *blas = parent()->AsBlas(); in ThenBlasGemmStridedBatched() local
Dstream.cc1250 if (blas::BlasSupport *blas = stream->parent_->AsBlas()) { in Run() local
/external/eigen/blas/
DCMakeLists.txt13 add_custom_target(blas) target
/external/tensorflow/tensorflow/core/kernels/
Dgpu_utils.cc156 if (auto* blas = stream_exec->AsBlas()) { in LogConvAutotuneResults() local
203 if (auto* blas = stream_exec->AsBlas()) { in LogFusedConvForwardAutotuneResults() local
/external/deqp/external/vulkancts/modules/vulkan/ray_tracing/
DvktRayTracingAccelerationStructuresTests.cpp1563 for (auto& blas : bottomLevelAccelerationStructures) in runTest() local
2212 …de::SharedPtr<BottomLevelAccelerationStructure> blas = de::SharedPtr<BottomLevelAccelerationStr… in iterate() local
2541 for (auto& blas : bottomStructures) in iterate() local
2701 …auto blas = de::SharedPtr<BottomLevelAccelerationStructure>(makeBottomLevelAccelerationStructure()… in prepareTopAccelerationStructure() local
2714 …auto blas = de::SharedPtr<BottomLevelAccelerationStructure>(makeBottomLevelAccelerationStructure()… in prepareTopAccelerationStructure() local
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
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_algorithm_picker.cc405 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
1959 for (auto& blas : rayQueryBottomLevelAccelerationStructures) in iterate() local
DvktRayQueryAccelerationStructuresTests.cpp1063 for (auto& blas : bottomLevelAccelerationStructures) in fillCommandBuffer() local
2495 for (auto& blas : bottomLevelAccelerationStructures) in runTest() local
3171 …de::SharedPtr<BottomLevelAccelerationStructure> blas = de::SharedPtr<BottomLevelAccelerationStr… in iterate() local
DvktRayQueryCullRayFlagsTests.cpp1065 for (auto& blas : bottomLevelAccelerationStructures) in fillCommandBuffer() local
/external/deqp/external/vulkancts/framework/vulkan/
DvkRayTracingUtil.cpp1670 const auto& blas = *bottomLevelInstances[instanceNdx]; in updateInstanceBuffer() local
1994 const auto& blas = *m_bottomLevelInstances[instanceIndex]; in updateInstanceMatrix() local
2277 de::MovePtr<BottomLevelAccelerationStructure> blas = makeBottomLevelAccelerationStructure(); in createAndDeserializeBottoms() local
/external/deqp/external/vulkancts/modules/vulkan/binding_model/
DvktBindingValveMutableTests.cpp431 BLASPtr blas; member
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_blas.cc2392 gpu::ROCMBlas *blas = new gpu::ROCMBlas(rocm_executor); in initialize_rocblas() local
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_blas.cc3723 gpu::CUDABlas *blas = new gpu::CUDABlas(cuda_executor); in initialize_cublas() local