- Sort Score
- Num 10 results
- Language All
Results 1 - 10 of 44 for tensors (0.07 seconds)
-
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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)