Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for GetOrCreateSubStream (0.26 sec)

  2. github.com/tensorflow/tensorflow

    tensorflow/compiler/jit/xla_tpu_device.cc 

                                 : nullptr;
    se::Stream* const dst_device_to_device_stream =
        should_use_substream
            ? device_to_device_master_stream->GetOrCreateSubStream().value_or(
                  nullptr)
            : dst_xla_context->GetDeviceToDeviceStream();
    TF_RET_CHECK(dst_device_to_device_stream != nullptr);
    auto return_substream = gtl::MakeCleanup(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 22:53:47 UTC 2024
    - 20.9K bytes
    - Viewed (0)
Back to top