| /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
|
| /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/deqp/external/vulkancts/modules/vulkan/ray_tracing/ |
| D | vktRayTracingAccelerationStructuresTests.cpp | 1907 for (auto &blas : bottomLevelAccelerationStructures) in runTest() local
2658 de::SharedPtr<BottomLevelAccelerationStructure> blas = in iterate() local
2998 for (auto &blas : bottomStructures) in iterate() local
3175 …auto blas = de::SharedPtr<BottomLevelAccelerationStructure>(makeBottomLevelAccelerationStructure()… in prepareTopAccelerationStructure() local
3188 auto blas = in prepareTopAccelerationStructure() local
3589 BlasPtr blas(makeBottomLevelAccelerationStructure().release()); in prepareBottomAccStructures() local
3831 [](const BlasPtr &blas) { return *blas->getPtr(); }); in iterate()
3838 for (BlasPtr blas : bottoms) in iterate() local
3880 [](const BlasPtr &blas) { return *blas.get()->getPtr(); }); in iterate()
4705 BlasPtr blas(makeBottomLevelAccelerationStructure().release()); in iterate() local
[all …]
|
| D | vktRayTracingTraversalControlTests.cpp | 699 for (auto &blas : bottomLevelAccelerationStructures) in runTest() local
|
| D | vktRayTracingBuildTests.cpp | 302 void RayTracingBuildTestInstance::initBottomAccelerationStructure(BlasPtr blas, bool useGpuBuild, in initBottomAccelerationStructure()
|
| D | vktRayTracingPipelineFlagsTests.cpp | 860 auto blas = makeBottomLevelAccelerationStructure(); in createBottomLevelAccelerationStructs() local
882 auto blas = makeBottomLevelAccelerationStructure(); in createBottomLevelAccelerationStructs() local
|
| D | vktRayTracingBuiltinTests.cpp | 3508 std::vector<de::SharedPtr<BottomLevelAccelerationStructure>> &blas) in initTopAccelerationStructure()
3611 std::vector<de::SharedPtr<BottomLevelAccelerationStructure>> blas; in initBottomAccelerationStructures() local
4071 …std::vector<de::SharedPtr<BottomLevelAccelerationStructure>> blas = initBottomAccelerationStructur… in iterate() local
|
| D | vktRayTracingCallableShadersTests.cpp | 828 for (auto &blas : bottomLevelAccelerationStructures) in runTest() local
|
| D | vktRayTracingShaderBindingTableTests.cpp | 1055 for (auto &blas : bottomLevelAccelerationStructures) in runTest() 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_query/ |
| D | vktRayQueryTraversalControlTests.cpp | 1062 for (auto &blas : bottomLevelAccelerationStructures) in fillCommandBuffer() local
2000 for (auto &blas : rayQueryBottomLevelAccelerationStructures) in iterate() local
|
| D | vktRayQueryAccelerationStructuresTests.cpp | 1084 for (auto &blas : bottomLevelAccelerationStructures) in fillCommandBuffer() local
2570 for (auto &blas : bottomLevelAccelerationStructures) in runTest() local
3330 de::SharedPtr<BottomLevelAccelerationStructure> blas = in iterate() local
|
| D | vktRayQueryMiscTests.cpp | 831 de::MovePtr<BottomLevelAccelerationStructure> blas = makeBottomLevelAccelerationStructure(); in createAccStructs() local
|
| D | vktRayQueryCullRayFlagsTests.cpp | 1089 for (auto &blas : bottomLevelAccelerationStructures) in fillCommandBuffer() local
|
| /external/deqp/external/vulkancts/framework/vulkan/ |
| D | vkRayTracingUtil.cpp | 1935 auto blas = new BottomLevelAccelerationStructurePoolMember(*m_impl); in add() local
2607 const auto &blas = *bottomLevelInstances[instanceNdx]; in updateInstanceBuffer() local
3032 const auto &blas = *m_bottomLevelInstances[instanceIndex]; in updateInstanceMatrix() local
3342 … 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/binding_model/ |
| D | vktBindingMutableTests.cpp | 530 BLASPtr blas; member
|
| /external/deqp/external/vulkancts/modules/vulkan/pipeline/ |
| D | vktPipelineShaderModuleIdentifierTests.cpp | 1992 BLASPtr blas; in iterate() local
|