Home
last modified time | relevance | path

Searched refs:GpuStreamMemberHack (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/host/
Dhost_stream.h38 void **GpuStreamMemberHack() override { return nullptr; } in GpuStreamMemberHack() function
/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_stream.h44 void** GpuStreamMemberHack() override { in GpuStreamMemberHack() function
/external/tensorflow/tensorflow/contrib/tensorrt/custom_plugin_examples/
Dinc_op_kernel.cu.cc70 ->GpuStreamMemberHack())); in Compute()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcusolver_context.cc97 stream->implementation()->GpuStreamMemberHack())); in Create()
Dnccl_all_reduce_thunk.cc287 participant.stream->implementation()->GpuStreamMemberHack()); in DoAllReduce()
/external/tensorflow/tensorflow/core/util/
Dcuda_launch_config.h298 ->GpuStreamMemberHack())); in GetCudaStream()
/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_internal.h135 virtual void **GpuStreamMemberHack() { return nullptr; } in GpuStreamMemberHack() function
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
Dtrt_engine_op.cc327 ->GpuStreamMemberHack())); in ExecuteCalibration()
503 ->GpuStreamMemberHack())); in ExecuteTrtEngine()
/external/tensorflow/tensorflow/core/nccl/
Dnccl_manager.cc545 comm_stream->implementation()->GpuStreamMemberHack()); in LoopKernelLaunches()
/external/tensorflow/tensorflow/core/kernels/
Dcuda_solvers.cc154 ->GpuStreamMemberHack())); in CudaSolver()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.cc937 streams_[stream_id]->compute->implementation()->GpuStreamMemberHack()); in ReinitializeDevice()