Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for scatter (0.32 sec)

  1. tensorflow/c/eager/c_api.h

    // a copy, returns the destination device of the copy. Note that the returned
    // device name is not always the device holding the tensor handle's memory. If
    // you want the latter, use TFE_TensorHandleBackingDeviceName. This function
    // will block till the operation that produces `h` has completed.
    TF_CAPI_EXPORT extern const char* TFE_TensorHandleDeviceName(
        TFE_TensorHandle* h, TF_Status* status);
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Apr 27 21:07:00 GMT 2023
    - 22.8K bytes
    - Viewed (1)
  2. tensorflow/c/eager/tape.h

      //
      // Like GradientTape::RecordOperation, this method takes the operation type
      // `op_type` (e.g. "Add"), the operation's inputs (`input_tensors`,
      // `input_tensor_id`, and `input_dtypes`; the latter two are somewhat
      // redundant but taken as arguments to avoid repeatedly fetching these values
      // between calls to ShouldRecord and Accumulator), and its outputs
      // (`output_tensors`).
      //
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Apr 02 12:40:29 GMT 2024
    - 47.2K bytes
    - Viewed (1)
  3. tensorflow/c/eager/parallel_device/parallel_device.cc

            // but need special handling for gradients (gradient of copy-on is not
            // just copy-off but includes a sum) and consideration of performance.
            //
            // TODO(allenl): There may be smarter ways to do this copy in some
            // cases, i.e. with a collective broadcast. We'll need to be careful
            // about things that are taken as inputs on the host or on their
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Mar 29 22:05:31 GMT 2023
    - 18.3K bytes
    - Viewed (0)
Back to top