Searched refs:host_outputs (Results 1 – 2 of 2) sorted by relevance
133 std::vector<std::vector<uint8>> host_outputs; in RunMlirModuleWithHostBuffers() local135 host_outputs.emplace_back(); in RunMlirModuleWithHostBuffers()136 host_outputs.back().resize(result.second.size()); in RunMlirModuleWithHostBuffers()137 stream->ThenMemcpy(static_cast<void*>(host_outputs.back().data()), in RunMlirModuleWithHostBuffers()141 return host_outputs; in RunMlirModuleWithHostBuffers()
201 Tensor& host_outputs(int i) { return host_outputs_[i]; } in host_outputs() function in tensorflow::__anon66d29a500511::EMBenchmarkHelper544 << bm_helper->host_outputs(0).SummarizeValue(64); in BM_chain_ops()