/external/tensorflow/tensorflow/stream_executor/ |
D | plugin_registry.h | 107 std::map<PluginId, BlasFactory> blas; member 117 PluginId blas, dnn, fft, rng; member
|
D | plugin.cc | 33 PluginConfig& PluginConfig::SetBlas(PluginId blas) { in SetBlas()
|
D | plugin.h | 76 PluginId blas() const { return blas_; } in blas() function
|
D | blas.cc | 21 namespace blas { namespace
|
D | stream.h | 1367 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
|
D | stream.cc | 1250 if (blas::BlasSupport *blas = stream->parent_->AsBlas()) { in Run() local
|
/external/eigen/blas/ |
D | CMakeLists.txt | 13 add_custom_target(blas) target
|
/external/tensorflow/tensorflow/core/kernels/ |
D | gpu_utils.cc | 156 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/ |
D | vktRayTracingAccelerationStructuresTests.cpp | 1563 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
|
D | vktRayTracingCallableShadersTests.cpp | 740 for (auto& blas : bottomLevelAccelerationStructures) in runTest() local
|
D | vktRayTracingTraversalControlTests.cpp | 662 for (auto& blas : bottomLevelAccelerationStructures) in runTest() local
|
D | vktRayTracingShaderBindingTableTests.cpp | 926 for (auto& blas : bottomLevelAccelerationStructures) in runTest() local
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | gpu_conv_algorithm_picker.cc | 405 if (auto* blas = stream_exec_->AsBlas()) { in PickBestAlgorithmNoCacheCuda() local
|
/external/deqp/external/vulkancts/modules/vulkan/ray_query/ |
D | vktRayQueryTraversalControlTests.cpp | 1042 for (auto& blas : bottomLevelAccelerationStructures) in fillCommandBuffer() local 1959 for (auto& blas : rayQueryBottomLevelAccelerationStructures) in iterate() local
|
D | vktRayQueryAccelerationStructuresTests.cpp | 1063 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
|
D | vktRayQueryCullRayFlagsTests.cpp | 1065 for (auto& blas : bottomLevelAccelerationStructures) in fillCommandBuffer() local
|
/external/deqp/external/vulkancts/framework/vulkan/ |
D | vkRayTracingUtil.cpp | 1670 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/ |
D | vktBindingValveMutableTests.cpp | 431 BLASPtr blas; member
|
/external/tensorflow/tensorflow/stream_executor/rocm/ |
D | rocm_blas.cc | 2392 gpu::ROCMBlas *blas = new gpu::ROCMBlas(rocm_executor); in initialize_rocblas() local
|
/external/tensorflow/tensorflow/stream_executor/cuda/ |
D | cuda_blas.cc | 3723 gpu::CUDABlas *blas = new gpu::CUDABlas(cuda_executor); in initialize_cublas() local
|