| /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/pytorch/aten/src/ATen/cuda/ |
| D | Exceptions.cpp | 7 namespace blas { namespace
|
| /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/deqp-deps/amber/src/vulkan/ |
| D | tlas.cc | 72 auto blas = instance->GetUsedBLAS(); in CreateTLAS() local
|
| D | blas.cc | 38 Result BLAS::CreateBLAS(amber::BLAS* blas) { in CreateBLAS()
|
| /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 | 1906 for (auto &blas : bottomLevelAccelerationStructures) in runTest() local
2655 de::SharedPtr<BottomLevelAccelerationStructure> blas = in iterate() local
2995 for (auto &blas : bottomStructures) in iterate() local
3172 …auto blas = de::SharedPtr<BottomLevelAccelerationStructure>(makeBottomLevelAccelerationStructure()… in prepareTopAccelerationStructure() local
3185 auto blas = in prepareTopAccelerationStructure() local
3586 BlasPtr blas(makeBottomLevelAccelerationStructure().release()); in prepareBottomAccStructures() local
3828 [](const BlasPtr &blas) { return *blas->getPtr(); }); in iterate()
3835 for (BlasPtr blas : bottoms) in iterate() local
3877 [](const BlasPtr &blas) { return *blas.get()->getPtr(); }); in iterate()
4702 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
|
| /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
|
| /external/deqp-deps/amber/src/ |
| D | acceleration_structure.h | 105 void SetUsedBLAS(const std::string& name, BLAS* blas) { in SetUsedBLAS()
|
| D | script.h | 220 Result AddBLAS(std::unique_ptr<BLAS> blas) { in AddBLAS()
|
| /external/deqp/external/vulkancts/modules/vulkan/ray_query/ |
| D | vktRayQueryMiscTests.cpp | 832 de::MovePtr<BottomLevelAccelerationStructure> blas = makeBottomLevelAccelerationStructure(); in createAccStructs() local
1135 const auto blas = blasPool.add(); in reuseScratchBufferInstance() local
1144 for (const auto &blas : blasPool.structures()) in reuseScratchBufferInstance() local
|
| D | vktRayQueryTraversalControlTests.cpp | 1062 for (auto &blas : bottomLevelAccelerationStructures) in fillCommandBuffer() local
2001 for (auto &blas : rayQueryBottomLevelAccelerationStructures) in iterate() local
|
| D | vktRayQueryAccelerationStructuresTests.cpp | 1084 for (auto &blas : bottomLevelAccelerationStructures) in fillCommandBuffer() local
2571 for (auto &blas : bottomLevelAccelerationStructures) in runTest() local
3331 de::SharedPtr<BottomLevelAccelerationStructure> blas = in iterate() local
|
| /external/deqp-deps/amber/src/amberscript/ |
| D | parser.cc | 4009 std::unique_ptr<BLAS> blas = MakeUnique<BLAS>(); in ParseBLAS() local
4066 Result Parser::ParseBLASTriangle(BLAS* blas) { in ParseBLASTriangle()
4115 Result Parser::ParseBLASAABB(BLAS* blas) { in ParseBLASAABB()
|
| /external/deqp/external/vulkancts/modules/vulkan/device_generated_commands/ |
| D | vktDGCRayTracingTestsExt.cpp | 258 auto blas = makeBottomLevelAccelerationStructure(); in makeBottomLevelASWithParams() local
319 TLASPtr makeTopLevelASWithParams(const std::vector<BLASPtr> &blas, const std::vector<CellParams> &c… in makeTopLevelASWithParams()
1004 std::vector<BLASPtr> blas; in iterate() local
|
| /external/deqp/external/vulkancts/framework/vulkan/ |
| D | vkRayTracingUtil.cpp | 1939 auto blas = new BottomLevelAccelerationStructurePoolMember(*m_impl); in add() local
2618 const auto &blas = *bottomLevelInstances[instanceNdx]; in updateInstanceBuffer() local
3044 const auto &blas = *m_bottomLevelInstances[instanceIndex]; in updateInstanceMatrix() local
3354 … de::MovePtr<BottomLevelAccelerationStructure> blas = makeBottomLevelAccelerationStructure(); in createAndDeserializeBottoms() local
|
| /external/deqp/external/vulkancts/modules/vulkan/pipeline/ |
| D | vktPipelineBinaryTests.cpp | 801 auto blas = makeBottomLevelAccelerationStructure(); in iterate() local
|