Search Options

Display Count
Sort
Preferred Language
Advanced Search

Results 1 - 10 of 62 for tensors (0.06 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.h

    TF_AbstractTensor* TF_OutputListGet(TF_OutputList* o, int i);
    // Append a tensor at the end of the output list, growing its size by one.
    void TF_OutputListPushBack(TF_OutputList* o, TF_AbstractTensor* tensor,
                               TF_Status*);
    
    // TF_ExecuteOperation will, if in eager mode, execute, if in graph mode, maybe
    // capture some inputs and then add a node in the graph. The output tensors are
    // returned through the provided TF_OutputList.
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Sun Oct 24 11:27:00 GMT 2021
    - 7K bytes
    - Click Count (0)
  4. 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)
  5. 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)
  6. 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)
  7. 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)
  8. tensorflow/c/c_api_macros.h

    #else
    #define TF_CAPI_EXPORT __attribute__((visibility("default")))
    #endif  // TF_CAPI_WEAK
    #endif  // _WIN32
    #endif  // SWIG
    
    // TF_Bool is the C API typedef for unsigned char, while TF_BOOL is
    // the datatype for boolean tensors.
    #ifndef TF_Bool
    #define TF_Bool unsigned char
    #endif  // TF_Bool
    
    // Macro used to calculate struct size for maintaining ABI stability across
    // different struct implementations.
    #ifndef TF_OFFSET_OF_END
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Sat May 13 04:44:45 GMT 2023
    - 1.6K bytes
    - Click Count (0)
  9. tensorflow/c/eager/c_api_debug_test.cc

      CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
    
      ASSERT_EQ(2, TFE_TensorDebugInfoOnDeviceNumDims(debug_info));
      // Shape is the same for CPU tensors.
      EXPECT_EQ(3, TFE_TensorDebugInfoOnDeviceDim(debug_info, 0));
      EXPECT_EQ(2, TFE_TensorDebugInfoOnDeviceDim(debug_info, 1));
    
      TFE_DeleteTensorDebugInfo(debug_info);
      TFE_DeleteTensorHandle(h);
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Mon Apr 06 22:10:09 GMT 2020
    - 2.3K bytes
    - Click Count (0)
  10. tensorflow/c/c_api.h

    // - Directly referencing external tensors from the cond/body graphs (this is
    //   possible in the Python API)
    TF_CAPI_EXPORT extern TF_WhileParams TF_NewWhile(TF_Graph* g, TF_Output* inputs,
                                                     int ninputs,
                                                     TF_Status* status);
    
    // Builds the while loop specified by `params` and returns the output tensors of
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Thu Oct 26 21:08:15 GMT 2023
    - 82.3K bytes
    - Click Count (0)
Back to Top