Home
last modified time | relevance | path

Searched refs:gputools (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dbatch_matmul_op_impl.h410 constexpr perftools::gputools::blas::Transpose kTranspose =
412 ? perftools::gputools::blas::Transpose::kConjugateTranspose
413 : perftools::gputools::blas::Transpose::kTranspose;
414 perftools::gputools::blas::Transpose trans[] = {
415 perftools::gputools::blas::Transpose::kNoTranspose, kTranspose};
426 typedef perftools::gputools::DeviceMemory<Scalar> DeviceMemoryType;
Dbias_op.cc568 perftools::gputools::Timer timer(stream->parent()); in Compute()
/external/tensorflow/tensorflow/
Dtf_exported_symbols.lds2 *perftools*gputools*
Dtf_version_script.lds4 *perftools*gputools*;
/external/tensorflow/tensorflow/stream_executor/
Dstream_executor.h39 namespace gputools {
Ddevice_memory.h32 namespace gputools {
/external/tensorflow/tensorflow/core/platform/
Dstream_executor.h36 namespace gputools {
/external/tensorflow/tensorflow/stream_executor/lib/
Dptr_util.h28 namespace gputools {
Darray_slice.h33 namespace gputools {
Dstatus.h50 namespace gputools {
/external/tensorflow/tensorflow/core/platform/default/gpu/
Dcupti_wrapper.h29 namespace gputools {
Dcupti_wrapper.cc26 namespace gputools { namespace
/external/tensorflow/tensorflow/compiler/xla/tests/
Dmultiple_devices_on_host_test.cc59 perftools::gputools::MultiPlatformManager::PlatformWithName("Host")); in TestWithDeviceCount()
/external/tensorflow/tensorflow/core/platform/default/
Ddevice_tracer.cc51 absl::make_unique<perftools::gputools::profiler::CuptiWrapper>()) { in ToStatus()
168 : cupti_wrapper_(new perftools::gputools::profiler::CuptiWrapper()), in CuptiCallbackHook()
315 std::unique_ptr<perftools::gputools::profiler::CuptiWrapper> cupti_wrapper_;
/external/tensorflow/tensorflow/tools/def_file_filter/
Ddef_file_filter.py.tpl70 r"perftools::gputools")
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h48 port::StatusOr<perftools::gputools::dnn::VersionInfo> GetVersion() override;
Dcuda_dnn.cc347 port::StatusOr<perftools::gputools::dnn::VersionInfo>
351 return perftools::gputools::dnn::VersionInfo( in GetVersion()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h46 port::StatusOr<perftools::gputools::dnn::VersionInfo> GetVersion() override;
Drocm_dnn.cc518 port::StatusOr<perftools::gputools::dnn::VersionInfo>
521 return perftools::gputools::dnn::VersionInfo(1, 3, 0); in GetVersion()