Search Options

Display Count
Sort
Preferred Language
Advanced Search

Results 1 - 10 of 11 for tensors (0.04 seconds)

The search processing time has exceeded the limit. The displayed results may be partial.

  1. tensorflow/c/eager/gradients.cc

          });
    }
    bool Tape::ShouldRecord(
        absl::Span<const AbstractTensorHandle* const> tensors) const {
      std::vector<int64_t> tensor_ids(tensors.size());
      std::vector<tensorflow::DataType> tensor_dtypes(tensors.size());
      for (int i = 0; i < tensors.size(); i++) {
        tensor_ids[i] = ToId(tensors[i]);
        tensor_dtypes[i] = tensors[i]->DataType();
      }
      return GradientTape::ShouldRecord(tensor_ids, tensor_dtypes);
    }
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Tue Feb 24 06:18:31 GMT 2026
    - 19.6K bytes
    - Click Count (0)
  2. tensorflow/c/eager/c_api.h

    // some input tensors are not on that device.
    // LINT.IfChange
    // Note: Keep in sync with internal copy of enum in eager/context.h.
    typedef enum TFE_ContextDevicePlacementPolicy {
      // Running operations with input tensors on the wrong device will fail.
      TFE_DEVICE_PLACEMENT_EXPLICIT = 0,
      // Copy the tensor to the right device but log a warning.
      TFE_DEVICE_PLACEMENT_WARN = 1,
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Thu Apr 27 21:07:00 GMT 2023
    - 22.8K bytes
    - Click Count (0)
  3. tensorflow/c/c_api_experimental.h

    // drop below the capacity due to dequeuing.
    //
    // Tensors are dequeued via the corresponding TF dequeue op.
    // TODO(hongm): Add support for `timeout_ms`.
    TF_CAPI_EXPORT extern void TF_EnqueueNamedTensor(TF_Session* session,
                                                     int tensor_id,
                                                     TF_Tensor* tensor,
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Thu Apr 27 21:07:00 GMT 2023
    - 15.1K bytes
    - Click Count (0)
  4. tensorflow/c/eager/immediate_execution_context.h

    // Note: Keep in sync with exported copy of enum in eager/c_api.h.
    enum ContextDevicePlacementPolicy {
      // Running operations with input tensors on the wrong device will fail.
      DEVICE_PLACEMENT_EXPLICIT = 0,
      // Copy the tensor to the right device but log a warning.
      DEVICE_PLACEMENT_WARN = 1,
      // Silently copy the tensor, which has a performance cost since the operation
      // will be blocked till the copy completes. This is the default policy.
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Sat Oct 12 05:11:17 GMT 2024
    - 12.3K bytes
    - Click Count (0)
  5. tensorflow/c/eager/dlpack.cc

        status->status = tensorflow::errors::InvalidArgument(
            "DLPack doesn't support ", handle->TypeString(), " tensor");
        return nullptr;
      }
      const tensorflow::Tensor* tensor;
      status->status = handle->Tensor(&tensor);
      if (!status->status.ok()) {
        return nullptr;
      }
      return tensor;
    }
    
    // Deleter for DLManagedTensor
    void DLManagedTensorDeleter(DLManagedTensor* arg) {
      TfDlManagedTensorCtx* owner =
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Thu Mar 13 23:41:52 GMT 2025
    - 13K bytes
    - Click Count (0)
  6. tensorflow/c/eager/c_api_experimental.h

      // shape is explicitly requested where possible.
      int (*num_dims)(void* data, TF_Status* status);
    
      // Computes the axis length at `dim_index`.
      int64_t (*dim)(void* data, int dim_index, TF_Status* status);
    
      void (*deallocator)(void* data);
    
      // Summarizes the value of this tensor. The caller takes ownership of the
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Wed Feb 21 22:37:46 GMT 2024
    - 39.5K bytes
    - Click Count (1)
  7. tensorflow/c/eager/c_api.cc

            "TFE_TensorHandleDeviceMemorySize may not be called on a ",
            handle->TypeString(), " tensor handle.");
        return 0;
      }
      const tensorflow::Tensor* tensor;
      status->status = handle->Tensor(&tensor);
      if (!status->status.ok()) {
        return 0;
      }
      return tensor->TotalBytes();
    }
    
    TFE_Op* TFE_NewOp(TFE_Context* ctx, const char* op_or_function_name,
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Fri Nov 07 05:55:21 GMT 2025
    - 43.9K bytes
    - Click Count (0)
  8. tensorflow/c/c_api_experimental.cc

      if (VLOG_IS_ON(1) && status->status.ok()) {
        tensorflow::Tensor tensor;
        if (tensorflow::TF_TensorToTensor(ret, &tensor).ok()) {
          VLOG(1) << "Dequeued tensor content: " << tensor.DebugString();
        }
      }
      return ret;
    }
    
    void TF_EnqueueNamedTensor(TF_Session* session, int tensor_id,
                               TF_Tensor* tensor, TF_Status* status) {
      assert(session);
      {
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Sat Oct 04 05:55:32 GMT 2025
    - 29.4K bytes
    - Click Count (0)
  9. tensorflow/c/c_test_util.cc

                              const char* name) {
      unique_tensor_ptr tensor(BoolTensor(v), TF_DeleteTensor);
      return Const(tensor.get(), graph, s, name);
    }
    
    TF_Operation* ScalarConst(int32_t v, TF_Graph* graph, TF_Status* s,
                              const char* name) {
      unique_tensor_ptr tensor(Int32Tensor(v), TF_DeleteTensor);
      return Const(tensor.get(), graph, s, name);
    }
    
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Sat Oct 04 05:55:32 GMT 2025
    - 17.8K bytes
    - Click Count (1)
  10. tensorflow/c/eager/c_api_unified_experimental_graph.cc

        op_->node_builder.Attr(attr_name, func_name);
        return absl::OkStatus();
      }
      absl::Status SetAttrTensor(const char* attr_name,
                                 AbstractTensorInterface* tensor) override {
        return tensorflow::errors::Unimplemented(
            "SetAttrTensor has not been implemented yet.");
      }
      absl::Status SetAttrStringList(const char* attr_name,
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Sat May 31 07:13:41 GMT 2025
    - 15.7K bytes
    - Click Count (0)
Back to Top