Searched defs:compute_stream (Results 1 – 4 of 4) sorted by relevance
104 auto* compute_stream = c->op_device_context()->stream(); in ComputeAsync() local136 auto* compute_stream = c->op_device_context()->stream(); in ComputeAsync() local173 auto* compute_stream = c->op_device_context()->stream(); in ComputeAsync() local208 auto* compute_stream = c->op_device_context()->stream(); in ComputeAsync() local245 auto* compute_stream = c->op_device_context()->stream(); in ComputeAsync() local
124 auto* compute_stream = col_ctx_->op_ctx->op_device_context()->stream(); in Run() local
34 int32 compute_stream = -1; member
72 std::shared_ptr<se::Stream> compute_stream, in XlaDeviceContext()