Search Options

Display Count
Sort
Preferred Language
Advanced Search

Results 1 - 10 of 44 for tensors (0.07 seconds)

  1. tensorflow/c/eager/gradients.h

                           const string& op_name = "");
      // Returns whether any tensor in a list of tensors is being watched and has
      // a trainable dtype.
      bool ShouldRecord(
          absl::Span<const AbstractTensorHandle* const> tensors) const;
      // Unwatches this tensor on the tape. Mainly used for cleanup when deleting
      // eager tensors.
      void DeleteTrace(const AbstractTensorHandle*);
    
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Sat Oct 12 05:11:17 GMT 2024
    - 6.9K bytes
    - Click Count (0)
  2. 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)
  3. tensorflow/c/eager/c_api_unified_experimental_internal.h

      }
    };
    
    // An abstract operation describes an operation by its type, name, and
    // attributes. It can be "executed" by the context with some input tensors.
    // It is allowed to reusing the same abstract operation for multiple execution
    // on a given context, with the same or different input tensors.
    class TracingOperation : public AbstractOperation {
     protected:
      explicit TracingOperation(AbstractOperationKind kind)
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Sat Oct 12 05:11:17 GMT 2024
    - 5.2K 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/c_api_unified_experimental_test.cc

      TFE_DeleteContextOptions(opts);
    
      ASSERT_EQ(TF_OK, TF_GetCode(status.get())) << TF_Message(status.get());
    
      /* Want to test simple MatMul example with abstract tensors:
        [[1,2],   *  [[5,6],   =   [[19,22],
         [3,4]]       [7,8]]        [43,50]]
      */
    
      // Build 1st Matrix.
      int64_t dims[] = {2, 2};  // Matrices will be 2 x 2
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Sat Oct 12 05:11:17 GMT 2024
    - 39.1K bytes
    - Click Count (0)
  6. RELEASE.md

        *   Transpose now supports 6D tensors.
        *   Float LSTM now supports diagonal recurrent tensors: https://arxiv.org/abs/1903.08023
    
    *   `tf.experimental.dtensor`:
    
        *   Coordination service now works with `dtensor.initialize_accelerator_system`, and enabled by default.
        *   Add `tf.experimental.dtensor.is_dtensor` to check if a tensor is a DTensor instance.
    
    *   `tf.data`:
    
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Mon Mar 30 18:31:38 GMT 2026
    - 746.5K bytes
    - Click Count (3)
  7. tensorflow/c/c_api_experimental_test.cc

      TFE_DeleteOp(matmul_op);
      // TODO(bgogul): Add some death tests where status is not OK.
    }
    
    TEST_F(ShapeInferenceTest, InfersShapesFromInputTensors) {
      // Prepare some tensors for shape.
      TF_Tensor* tensor_1X6 = Int32Tensor({1, 6});
      CHECK_EQ(TF_OK, TF_GetCode(status_)) << TF_Message(status_);
      TF_Tensor* tensor_1X1X6 = Int32Tensor({1, 1, 6});
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Wed Jan 07 04:56:09 GMT 2026
    - 13.1K bytes
    - Click Count (0)
  8. tensorflow/c/c_api.cc

        // Input tensors
        const std::vector<std::pair<string, Tensor>>& input_pairs,
        // Output tensors
        const std::vector<string>& output_tensor_names, TF_Tensor** c_outputs,
        // Target nodes
        const std::vector<string>& target_oper_names, TF_Buffer* run_metadata,
        TF_Status* status) {
      const int noutputs = output_tensor_names.size();
      std::vector<Tensor> outputs(noutputs);
      Status result;
    
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Sat Oct 04 05:55:32 GMT 2025
    - 102.4K bytes
    - Click Count (0)
  9. 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)
  10. 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)
Back to Top