Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for CreateLocalHandle (0.24 sec)

  1. tensorflow/c/experimental/gradients/nn_grad.cc

      }
      auto imm_ctx = dyn_cast<ImmediateExecutionContext>(ctx);
      AbstractTensorPtr minus_1(imm_ctx->CreateInt32Scalar(-1));
      ImmediateTensorHandlePtr dim(imm_ctx->CreateLocalHandle(minus_1.get()));
      AbstractTensorHandle* expand_dims_outputs;
      TF_RETURN_IF_ERROR(
          ops::ExpandDims(ctx, vec, dim.get(), &expand_dims_outputs, "ExpandDims"));
      TF_RETURN_IF_ERROR(
    C++
    - Registered: Tue Mar 26 12:39:09 GMT 2024
    - Last Modified: Wed Feb 28 13:53:47 GMT 2024
    - 5.7K bytes
    - Viewed (0)
  2. tensorflow/c/eager/immediate_execution_context.h

          MemoryReleaser memory_releaser, void* memory_releaser_arg) = 0;
    
      // Create a handle to wrap and manage a Tensor
      virtual ImmediateExecutionTensorHandle* CreateLocalHandle(
          AbstractTensorInterface* t) = 0;
      // Copy the handle to another device.
      virtual ImmediateExecutionTensorHandle* CopyTensorHandleToDevice(
          ImmediateExecutionTensorHandle* handle, const char* device_name,
    C
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Thu Jul 06 08:34:00 GMT 2023
    - 12.3K bytes
    - Viewed (0)
  3. tensorflow/c/eager/c_api.cc

      tensorflow::Tensor tensor;
      status->status = tensorflow::TF_TensorToTensor(t, &tensor);
      if (!status->status.ok()) return nullptr;
    
      return tensorflow::wrap(tensorflow::TensorHandle::CreateLocalHandle(tensor));
    }
    
    void TFE_DeleteTensorHandle(TFE_TensorHandle* h) {
      if (h == nullptr) return;
    
      tensorflow::profiler::TraceMe activity(
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 43.9K bytes
    - Viewed (2)
  4. tensorflow/c/c_api_experimental.cc

      std::memcpy(tensorflow::TensorCApi::Buffer(tensor)->data(), data, len);
    
      status->status = absl::OkStatus();
      return tensorflow::wrap(tensorflow::TensorHandle::CreateLocalHandle(tensor));
    }
    
    // Set server_def on the context, possibly updating it.
    TF_CAPI_EXPORT extern void TFE_EnableCollectiveOps(TFE_Context* ctx,
                                                       const void* proto,
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 29.4K bytes
    - Viewed (0)
  5. tensorflow/c/eager/c_api_experimental.cc

    }
    
    TFE_TensorHandle* TFE_NewTensorHandleFromTensor(TFE_Context* ctx, TF_Tensor* t,
                                                    TF_Status* status) {
      return tensorflow::wrap(
          tensorflow::unwrap(ctx)->CreateLocalHandle(t->tensor));
    }
    
    TFE_TensorHandle* TFE_CreatePackedTensorHandle(TFE_Context* ctx,
                                                   TFE_TensorHandle** handles,
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Thu Apr 11 23:52:39 GMT 2024
    - 35.9K bytes
    - Viewed (3)
Back to top