Home
last modified time | relevance | path

Searched defs:se_stream (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/tpu/
Dtpu_stream.h76 SE_Stream* se_stream() const { return stream_; } in se_stream() function
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/
Dtf_rocm_runtime_wrappers.cc104 stream_executor::Stream *se_stream = ctx->op_device_context()->stream(); in tfKernelGenLaunchKernel() local
Dtf_cuda_runtime_wrappers.cc103 stream_executor::Stream *se_stream = ctx->op_device_context()->stream(); in tfKernelGenLaunchKernel() local