- Sort Score
- Num 10 results
- Language All
Results 1 - 3 of 3 for tensor_ids (0.05 seconds)
-
tensorflow/c/eager/gradients.cc
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); } void Tape::DeleteTrace(const AbstractTensorHandle* t) {
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/c_api_internal.h
// ExtendSessionGraphHelper manually. std::atomic<bool> extend_before_run; }; struct TF_ImportGraphDefOptions { tensorflow::ImportGraphDefOptions opts; // Backing memory for TensorId fields in opts. // TODO(skyewm): it'd be better if ImportGraphDefOptions owned this. std::vector<std::string> tensor_id_data; }; struct TF_ImportGraphDefResults { std::vector<TF_Output> return_tensors;
Created: Tue Apr 07 12:39:13 GMT 2026 - Last Modified: Wed Jan 07 04:56:09 GMT 2026 - 7.5K bytes - Click Count (0) -
tensorflow/c/c_api_test.cc
#include "tensorflow/core/framework/tensor.h" #include "tensorflow/core/framework/tensor.pb.h" #include "tensorflow/core/framework/tensor_shape.pb.h" #include "tensorflow/core/framework/types.pb.h" #include "tensorflow/core/graph/tensor_id.h" #include "tensorflow/core/lib/core/status_test_util.h" #include "tensorflow/core/lib/io/path.h" #include "tensorflow/core/platform/path.h" #include "tensorflow/core/platform/protobuf.h"
Created: Tue Apr 07 12:39:13 GMT 2026 - Last Modified: Wed Jan 07 04:56:09 GMT 2026 - 97.3K bytes - Click Count (0)