| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | host_callback.h | 69 struct HostCallback { struct
71 std::vector<HostCallbackArgInfo> operands; argument
72 std::vector<HostCallbackArgInfo> results; argument
86 HostCallbackContext(HostCallback host_callback, PjRtClient* client) in HostCallbackContext() argument
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| D | stream_executor_internal.cc | 23 bool StreamExecutorInterface::HostCallback(Stream* stream, in HostCallback() function in stream_executor::internal::StreamExecutorInterface
|
| D | stream_executor_pimpl.cc | 735 bool StreamExecutor::HostCallback(Stream* stream, in HostCallback() function in stream_executor::StreamExecutor
740 bool StreamExecutor::HostCallback(Stream* stream, in HostCallback() function in stream_executor::StreamExecutor
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executor.cc | 98 bool XlaInterpreterExecutor::HostCallback( in HostCallback() function in stream_executor::interpreter::XlaInterpreterExecutor
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/host/ |
| D | host_gpu_executor.cc | 183 bool HostExecutor::HostCallback(Stream* stream, in HostCallback() function in stream_executor::host::HostExecutor
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/ |
| D | tpu_executor.cc | 358 bool TpuExecutor::HostCallback(Stream* stream, in HostCallback() function in tensorflow::tpu::TpuExecutor
|
| /external/tensorflow/tensorflow/c/experimental/stream_executor/ |
| D | stream_executor_test_util.cc | 83 TF_Bool HostCallback(const SP_Device* const device, SP_Stream stream, in HostCallback() function
|
| D | stream_executor.cc | 426 bool HostCallback(Stream* stream, in HostCallback() function in stream_executor::__anoncabb94000111::CStreamExecutor
|
| /external/rust/crates/grpcio-sys/grpc/third_party/cares/cares/test/ |
| D | ares-test.cc | 556 void HostCallback(void *data, int status, int timeouts, in HostCallback() function
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_gpu_executor.cc | 561 bool GpuExecutor::HostCallback(Stream* stream, in HostCallback() function in stream_executor::gpu::GpuExecutor
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_gpu_executor.cc | 693 bool GpuExecutor::HostCallback(Stream* stream, in HostCallback() function in stream_executor::gpu::GpuExecutor
|