Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for pushStack (0.12 sec)

  1. tensorflow/c/eager/c_api_remote_test_util.cc

        for (int i = 0; i < 50; ++i) {
          handles_task0.push_back(TestMatrixTensorHandle(ctx));
        }
      }
      const char task1_name[] = "/job:localhost/replica:0/task:1/device:CPU:0";
      const char task2_name[] = "/job:localhost/replica:0/task:2/device:CPU:0";
    
      std::vector<TFE_TensorHandle*> handles_task2;
      for (auto* h_task0 : handles_task0) {
        handles_task2.push_back(
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Fri Dec 11 22:56:03 UTC 2020
    - 9.1K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api_debug.cc

        return shape;
      }
      shape.reserve(rank);
      for (int i = 0; i < rank; ++i) {
        int64_t dim;
        *status = handle.Dim(i, &dim);
        if (!status->ok()) {
          return shape;
        }
        shape.push_back(dim);
      }
      return shape;
    }
    
    }  // namespace
    
    extern "C" {
    
    TF_CAPI_EXPORT extern TFE_TensorDebugInfo* TFE_TensorHandleTensorDebugInfo(
        TFE_TensorHandle* h, TF_Status* status) {
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 2.5K bytes
    - Viewed (0)
  3. tensorflow/c/eager/custom_device_testutil.cc

      TFE_DeleteOp(op);
      if (TF_GetCode(s) != TF_OK) return;
      std::vector<TFE_TensorHandle*> unwrapped_outputs;
      unwrapped_outputs.reserve(op_outputs.size());
      for (auto* handle : op_outputs) {
        unwrapped_outputs.push_back(handle);
      }
      for (int i = 0; i < *num_outputs; ++i) {
        auto logged_tensor = std::make_unique<LoggedTensor>(unwrapped_outputs[i]);
        outputs[i] = MakeLoggedTensorHandle(context, dev->device_name,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Wed Mar 03 20:47:31 UTC 2021
    - 8.3K bytes
    - Viewed (0)
  4. tensorflow/c/eager/c_api_unified_experimental.cc

      return wrap(unwrap(o)->outputs[i]);
    }
    void TF_OutputListPushBack(TF_OutputList* o, TF_AbstractTensor* tensor,
                               TF_Status* s) {
      unwrap(o)->outputs.push_back(unwrap(tensor));
    }
    
    void TF_AbstractOpSetOpType(TF_AbstractOp* op, const char* const op_type,
                                TF_Status* s) {
      tsl::Set_TF_Status_from_Status(
          s, unwrap(op)->Reset(op_type,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 9K bytes
    - Viewed (0)
Back to top