Searched defs:compute_stream (Results 1 – 6 of 6) sorted by relevance
108 auto* compute_stream = c->op_device_context()->stream(); in ComputeAsync() local139 auto* compute_stream = c->op_device_context()->stream(); in ComputeAsync() local175 auto* compute_stream = c->op_device_context()->stream(); in ComputeAsync() local209 auto* compute_stream = c->op_device_context()->stream(); in ComputeAsync() local245 auto* compute_stream = c->op_device_context()->stream(); in ComputeAsync() local
106 se::Stream* compute_stream() const { return compute_stream_.get(); } in compute_stream() function
78 auto* compute_stream = col_ctx->op_ctx->op_device_context()->stream(); in Enqueue() local
77 std::shared_ptr<se::Stream> compute_stream, in XlaDeviceContext()
202 se::Stream* compute_stream, in GPUKernelTracker()
159 se::Stream* compute_stream() { return gpu_->stream_->compute; } in compute_stream() function in tensorflow::GPUDeviceTestHelper