| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| H A D | plugin_registry.h | 107 std::map<PluginId, BlasFactory> blas; member
117 PluginId blas, dnn, fft, rng; member
|
| H A 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
|
| H A D | plugin.cc | 33 PluginConfig& PluginConfig::SetBlas(PluginId blas) { in SetBlas()
|
| H A D | plugin.h | 76 PluginId blas() const { return blas_; } in blas() function
|
| H A D | blas.cc | 21 namespace blas { namespace
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | gpu_utils.cc | 168 if (auto* blas = stream_exec->AsBlas()) { in LogConvAutotuneResults() local
215 if (auto* blas = stream_exec->AsBlas()) { in LogFusedConvForwardAutotuneResults() local
|
| /aosp_15_r20/external/deqp/external/vulkancts/modules/vulkan/ray_tracing/ |
| H A 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 …]
|
| H A D | vktRayTracingPipelineFlagsTests.cpp | 860 auto blas = makeBottomLevelAccelerationStructure(); in createBottomLevelAccelerationStructs() local
882 auto blas = makeBottomLevelAccelerationStructure(); in createBottomLevelAccelerationStructs() local
|
| H A D | vktRayTracingTraversalControlTests.cpp | 699 for (auto &blas : bottomLevelAccelerationStructures) in runTest() local
|
| H A D | vktRayTracingBuildTests.cpp | 302 void RayTracingBuildTestInstance::initBottomAccelerationStructure(BlasPtr blas, bool useGpuBuild, in initBottomAccelerationStructure()
|
| H A 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
|
| H A D | vktRayTracingCallableShadersTests.cpp | 828 for (auto &blas : bottomLevelAccelerationStructures) in runTest() local
|
| H A D | vktRayTracingShaderBindingTableTests.cpp | 1055 for (auto &blas : bottomLevelAccelerationStructures) in runTest() local
|
| /aosp_15_r20/cts/tests/tests/rsblas/src/android/renderscript/cts/ |
| H A D | BNNMTest.java | 145 ScriptIntrinsicBLAS blas = ScriptIntrinsicBLAS.create(mRS); in runBNNM() local
463 ScriptIntrinsicBLAS blas = ScriptIntrinsicBLAS.create(mRS); in testExceptionHandling() local
|
| /aosp_15_r20/external/eigen/blas/ |
| H A D | CMakeLists.txt | 13 add_custom_target(blas) target
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/cuda/ |
| H A D | Exceptions.cpp | 7 namespace blas { namespace
|
| H A D | CUDABlas.cpp | 212 namespace at::cuda::blas { namespace
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A 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
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A D | cuda_blas_lt.cc | 309 CUDABlas* blas = dynamic_cast<CUDABlas*>(stream->parent()->AsBlas()); in GetBlasLt() local
|
| /aosp_15_r20/external/deqp/external/vulkancts/modules/vulkan/ray_query/ |
| H A D | vktRayQueryTraversalControlTests.cpp | 1062 for (auto &blas : bottomLevelAccelerationStructures) in fillCommandBuffer() local
2000 for (auto &blas : rayQueryBottomLevelAccelerationStructures) in iterate() local
|
| H A 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
|
| H A D | vktRayQueryMiscTests.cpp | 831 de::MovePtr<BottomLevelAccelerationStructure> blas = makeBottomLevelAccelerationStructure(); in createAccStructs() local
|
| H A D | vktRayQueryCullRayFlagsTests.cpp | 1089 for (auto &blas : bottomLevelAccelerationStructures) in fillCommandBuffer() local
|
| /aosp_15_r20/external/deqp/external/vulkancts/framework/vulkan/ |
| H A D | vkRayTracingUtil.cpp | 1936 auto blas = new BottomLevelAccelerationStructurePoolMember(*m_impl); in add() local
2608 const auto &blas = *bottomLevelInstances[instanceNdx]; in updateInstanceBuffer() local
3034 const auto &blas = *m_bottomLevelInstances[instanceIndex]; in updateInstanceMatrix() local
3344 … de::MovePtr<BottomLevelAccelerationStructure> blas = makeBottomLevelAccelerationStructure(); in createAndDeserializeBottoms() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/ |
| H A D | rocm_blas.cc | 1189 gpu::ROCMBlas *blas = new gpu::ROCMBlas(rocm_executor); in initialize_rocblas() local
|