- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 22 for int64 (0.14 sec)
-
tensorflow/c/c_api.cc
int64_t value) { desc->node_builder.Attr(attr_name, static_cast<int64_t>(value)); } void TF_SetAttrIntList(TF_OperationDescription* desc, const char* attr_name, const int64_t* values, int num_values) { desc->node_builder.Attr( attr_name, ArraySlice<const int64_t>( reinterpret_cast<const int64_t*>(values), num_values)); }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 102.3K bytes - Viewed (0) -
tensorflow/c/eager/dlpack_test.cc
#include "tensorflow/core/platform/test.h" namespace tensorflow { namespace { void TestHandleFromDLPack(TF_Status* status, TFE_Context* ctx, std::vector<int64_t> shape, std::vector<int64_t> strides) { size_t num_elements = 1; for (int i = 0; i < static_cast<int32_t>(shape.size()); ++i) { num_elements *= shape[i]; } std::vector<float> data(num_elements);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Jun 30 03:04:46 GMT 2023 - 4.4K bytes - Viewed (0) -
tensorflow/c/experimental/filesystem/plugins/posix/posix_filesystem.cc
// `pread` returns a `ssize_t` on POSIX, but due to interface being // cross-platform, return type of `Read` is `int64_t`. int64_t r = int64_t{pread(posix_file->fd, dst, requested_read_length, static_cast<off_t>(offset))}; if (r > 0) { dst += r; offset += static_cast<uint64_t>(r); n -= r; // safe as 0 < r <= n so n will never underflow read += r; } else if (r == 0) {
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Sun Mar 24 20:08:23 GMT 2024 - 15.8K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/grad_test_helper.cc
} } void CheckTensorValue(AbstractTensorHandle* t, absl::Span<const float> manuals, absl::Span<const int64_t> dims, double abs_error) { TF_Tensor* analytical_tensor; auto s = GetValue(t, &analytical_tensor); ASSERT_EQ(errors::OK, s.code()) << s.message(); int64_t num_elem_analytical = 1; auto num_dims_analytical = TF_NumDims(analytical_tensor); ASSERT_EQ(dims.size(), num_dims_analytical);
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 5K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental.cc
} tensorflow::PartialTensorShape partial_shape; if (shape.num_dims != -1) { DCHECK(shape.dim_sizes != nullptr); Status status = tensorflow::PartialTensorShape::MakePartialShape( reinterpret_cast<int64_t*>(shape.dim_sizes), shape.num_dims, &partial_shape); if (!status.ok()) { tsl::Set_TF_Status_from_Status(s, status); return nullptr; } } tsl::Set_TF_Status_from_Status(
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 9K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.cc
const int64_t* dims, int num_dims) { DCHECK(index >= 0 && index < shape_list->num_items); TF_ShapeAndType& shape = shape_list->items[index]; DCHECK(shape.dims == nullptr) << "Shape at " << index << " is already set!"; DCHECK(num_dims >= 0) << "Number of dimensions cannot be negative!"; shape.num_dims = num_dims; shape.dims = new int64_t[num_dims];
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 29.4K bytes - Viewed (0) -
tensorflow/c/eager/gradients.cc
TapeVSpace vspace(ctx); std::vector<int64_t> target_tensor_ids = MakeTensorIDList(targets); std::vector<int64_t> source_tensor_ids = MakeTensorIDList(sources); tensorflow::gtl::FlatSet<int64_t> sources_set(source_tensor_ids.begin(), source_tensor_ids.end()); std::unordered_map<int64_t, TapeTensor> sources_that_are_targets;
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 19.3K bytes - Viewed (0) -
tensorflow/c/eager/dlpack.cc
} // Checks whether the stride array matches the layout of compact, row-majored // data. bool IsValidStrideCompactRowMajorData(int64_t* shape_arr, int64_t* stride_arr, int ndim) { bool valid = true; int64_t expected_stride = 1; for (int i = ndim - 1; i >= 0; --i) { // Empty tensors are always compact regardless of strides. if (shape_arr[i] == 0) return true;
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 12.8K bytes - Viewed (0) -
tensorflow/c/experimental/next_pluggable_device/c_api.cc
auto* cc_agent = reinterpret_cast<tsl::CoordinationServiceAgent*>(agent); return cc_agent->IsInitialized(); } void TF_CoordinationServiceInsertKeyValue(const char* key, int64_t key_size, const char* value, int64_t value_size, TF_CoordinationServiceAgent* agent, TF_Status* status) {
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Tue Jan 09 00:52:04 GMT 2024 - 13.9K bytes - Viewed (1) -
tensorflow/c/eager/parallel_device/parallel_device_lib.cc
std::vector<TFE_TensorHandle*> inputs, const TFE_OpAttrs* attributes, int expected_max_outputs, CancellationManager& cancellation_manager, absl::optional<int64_t> step_id = absl::nullopt); // Block until the previous `StartExecute` operation has executed. Forwards // the status from `TFE_Execute` and returns outputs if the status is OK.
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Feb 09 07:47:20 GMT 2024 - 25.4K bytes - Viewed (1)