| /external/tensorflow/tensorflow/compiler/xla/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 | 796 blas::BlasSupport *blas = parent()->AsBlas(); in ThenBlasGemm() local
891 blas::BlasSupport *blas = parent()->AsBlas(); in ThenBlasGemmWithAlgorithm() local
930 blas::BlasSupport *blas = parent()->AsBlas(); in ThenBlasGemmStridedBatchedWithAlgorithm() local
1048 blas::BlasSupport *blas = parent()->AsBlas(); in ThenBlasGemmStridedBatched() local
|
| D | stream.cc | 1084 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 | 168 if (auto* blas = stream_exec->AsBlas()) { in LogConvAutotuneResults() local
215 if (auto* blas = stream_exec->AsBlas()) { in LogFusedConvForwardAutotuneResults() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_algorithm_picker.cc | 569 if (auto* blas = stream_exec_->AsBlas()) { in AutotuneOneConvRunner() local
693 if (auto* blas = stream_exec_->AsBlas()) { in PickBestAlgorithmNoCacheCuda() local
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_blas_lt.cc | 309 CUDABlas* blas = dynamic_cast<CUDABlas*>(stream->parent()->AsBlas()); in GetBlasLt() local
|
| D | cuda_blas.cc | 1685 CUDABlas *blas = new CUDABlas(cuda_executor); in DoBlasInternalImpl() local
|
| /external/deqp/external/vulkancts/modules/vulkan/ray_tracing/ |
| D | vktRayTracingAccelerationStructuresTests.cpp | 1674 for (auto& blas : bottomLevelAccelerationStructures) in runTest() local
2323 …de::SharedPtr<BottomLevelAccelerationStructure> blas = de::SharedPtr<BottomLevelAccelerationStr… in iterate() local
2652 for (auto& blas : bottomStructures) in iterate() local
2812 …auto blas = de::SharedPtr<BottomLevelAccelerationStructure>(makeBottomLevelAccelerationStructure()… in prepareTopAccelerationStructure() local
2825 …auto blas = de::SharedPtr<BottomLevelAccelerationStructure>(makeBottomLevelAccelerationStructure()… in prepareTopAccelerationStructure() local
3191 BlasPtr blas(makeBottomLevelAccelerationStructure().release()); in prepareBottomAccStructures() local
3410 …std::transform(bottoms.begin(), bottoms.end(), std::next(handles.begin()), [](const BlasPtr& blas)… in iterate()
3417 for (BlasPtr blas : bottoms) sourceStructures.push_back(makeASInterfacePtr(blas)); in iterate() local
3454 …std::transform(bottoms.begin(), bottoms.end(), std::next(handles.begin()), [](const BlasPtr& blas)… in iterate()
4193 BlasPtr blas (makeBottomLevelAccelerationStructure().release()); in iterate() local
|
| D | vktRayTracingCallableShadersTests.cpp | 740 for (auto& blas : bottomLevelAccelerationStructures) in runTest() local
|
| D | vktRayTracingTraversalControlTests.cpp | 662 for (auto& blas : bottomLevelAccelerationStructures) in runTest() local
|
| D | vktRayTracingBuildTests.cpp | 310 void RayTracingBuildTestInstance::initBottomAccelerationStructure (BlasPtr blas, in initBottomAccelerationStructure()
|
| D | vktRayTracingPipelineFlagsTests.cpp | 821 auto blas = makeBottomLevelAccelerationStructure(); in createBottomLevelAccelerationStructs() local
841 auto blas = makeBottomLevelAccelerationStructure(); in createBottomLevelAccelerationStructs() local
|
| D | vktRayTracingShaderBindingTableTests.cpp | 926 for (auto& blas : bottomLevelAccelerationStructures) in runTest() 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
2520 for (auto& blas : bottomLevelAccelerationStructures) in runTest() local
3196 …de::SharedPtr<BottomLevelAccelerationStructure> blas = de::SharedPtr<BottomLevelAccelerationStr… in iterate() local
|
| D | vktRayQueryMiscTests.cpp | 828 de::MovePtr<BottomLevelAccelerationStructure> blas = makeBottomLevelAccelerationStructure(); in createAccStructs() local
|
| D | vktRayQueryCullRayFlagsTests.cpp | 1065 for (auto& blas : bottomLevelAccelerationStructures) in fillCommandBuffer() local
|
| /external/deqp/external/vulkancts/framework/vulkan/ |
| D | vkRayTracingUtil.cpp | 1730 auto blas = new BottomLevelAccelerationStructurePoolMember(*m_impl); in add() local
2401 const auto& blas = *bottomLevelInstances[instanceNdx]; in updateInstanceBuffer() local
2811 const auto& blas = *m_bottomLevelInstances[instanceIndex]; in updateInstanceMatrix() local
3094 de::MovePtr<BottomLevelAccelerationStructure> blas = makeBottomLevelAccelerationStructure(); in createAndDeserializeBottoms() local
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_blas.cc | 1189 gpu::ROCMBlas *blas = new gpu::ROCMBlas(rocm_executor); in initialize_rocblas() local
|
| /external/deqp/external/vulkancts/modules/vulkan/pipeline/ |
| D | vktPipelineShaderModuleIdentifierTests.cpp | 1923 BLASPtr blas; in iterate() local
|
| /external/deqp/external/vulkancts/modules/vulkan/binding_model/ |
| D | vktBindingMutableTests.cpp | 511 BLASPtr blas; member
|