Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for outputId (0.26 sec)

  1. tensorflow/c/eager/c_api_experimental.h

      // device if any of their inputs are on that custom device, but custom devices
      // are free to set a bad status in order to require explicit placement.
      void (*execute)(const TFE_Op* op, int* num_outputs,
                      TFE_TensorHandle** outputs, TF_Status* s, void* device_info);
    
      // Method to delete a device.
      void (*delete_device)(void* device_info);
    
      // Implements TFE_CreatePackedTensorHandle when one of `handles` is on this
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Feb 21 22:37:46 GMT 2024
    - 39.5K bytes
    - Viewed (0)
  2. tensorflow/c/eager/tape.h

      sources.reserve(output_tensors.size());
      for (const TapeTensor& output_tensor : output_tensors) {
        // Ownership of `aid` transferred to CallBackwardFunction below.
        Gradient* aid;
        if (output_tensor.GetDType() == tensorflow::DT_VARIANT) {
          // Note: Needs to be zeros rather than ones since there's currently no
          // ones_like for variants.
          aid = output_tensor.ZerosLike();
        } else {
    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/unified_api_testutil.h

    // being equivalent to the following python code.
    //
    // if use_function:
    //   outputs = tf.function(model)(inputs)
    // else:
    //   outputs = model(inputs)
    Status RunModel(Model model, AbstractContext* ctx,
                    absl::Span<AbstractTensorHandle* const> inputs,
                    absl::Span<AbstractTensorHandle*> outputs, bool use_function);
    
    Status BuildImmediateExecutionContext(bool use_tfrt, AbstractContext** ctx);
    
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Feb 27 13:57:45 GMT 2024
    - 4K bytes
    - Viewed (0)
Back to top