Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for model_out (0.06 sec)

  1. tensorflow/c/eager/gradient_checker.cc

      AbstractTensorHandlePtr model_out(model_outputs[0]);
    
      TF_Tensor* model_out_tensor;
      TF_RETURN_IF_ERROR(GetValue(model_out.get(), &model_out_tensor));
      int num_dims_out = TF_NumDims(model_out_tensor);
      TF_DeleteTensor(model_out_tensor);
    
      // If the output is a scalar, then return the scalar output
      if (num_dims_out == 0) {
        outputs[0] = model_out.release();
        return absl::OkStatus();
      }
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 7.3K bytes
    - Viewed (0)
Back to top