Searched refs:cross_program_prefetch_addr (Results 1 – 7 of 7) sorted by relevance
41 std::optional<se::DeviceMemoryBase> cross_program_prefetch_addr) { in LoadProgramAndEnqueueToStream() argument55 if (cross_program_prefetch_addr.has_value()) { in LoadProgramAndEnqueueToStream()56 prefetch_base = SE_DeviceMemoryBase{cross_program_prefetch_addr->opaque(), in LoadProgramAndEnqueueToStream()57 cross_program_prefetch_addr->size(), in LoadProgramAndEnqueueToStream()58 cross_program_prefetch_addr->payload()}; in LoadProgramAndEnqueueToStream()88 cross_program_prefetch_addr.has_value(); in LoadProgramAndEnqueueToStream()89 params.cross_program_prefetch_addr = in LoadProgramAndEnqueueToStream()90 cross_program_prefetch_addr.has_value() ? &prefetch_base : nullptr; in LoadProgramAndEnqueueToStream()
207 std::optional<se::DeviceMemoryBase> cross_program_prefetch_addr; in ExecuteAsyncOnStream() local225 cross_program_prefetch_addr = it->second.AsDeviceMemoryBase(); in ExecuteAsyncOnStream()237 cross_program_prefetch_addr)); in ExecuteAsyncOnStream()
57 cross_program_prefetch_addr) override;
77 cross_program_prefetch_addr) = 0;
53 cross_program_prefetch_addr) override { in LoadProgramAndEnqueueToStream()
41 absl::optional<se::DeviceMemoryBase> cross_program_prefetch_addr) { in LoadProgramAndEnqueueToStream() argument55 if (cross_program_prefetch_addr.has_value()) { in LoadProgramAndEnqueueToStream()56 prefetch_base = SE_DeviceMemoryBase{cross_program_prefetch_addr->opaque(), in LoadProgramAndEnqueueToStream()57 cross_program_prefetch_addr->size(), in LoadProgramAndEnqueueToStream()58 cross_program_prefetch_addr->payload()}; in LoadProgramAndEnqueueToStream()88 cross_program_prefetch_addr.has_value(); in LoadProgramAndEnqueueToStream()89 params.cross_program_prefetch_addr = in LoadProgramAndEnqueueToStream()90 cross_program_prefetch_addr.has_value() ? &prefetch_base : nullptr; in LoadProgramAndEnqueueToStream()
232 SE_DeviceMemoryBase* cross_program_prefetch_addr; member