- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 24 for dlim (0.13 sec)
-
tensorflow/c/c_api_internal.h
: #if !defined(IS_MOBILE_PLATFORM) && !defined(IS_SLIM_BUILD) api_def_map(op_list), #endif // !defined(IS_MOBILE_PLATFORM) && !defined(IS_SLIM_BUILD) update_docs_called(false) { } #if !defined(IS_MOBILE_PLATFORM) && !defined(IS_SLIM_BUILD) tensorflow::ApiDefMap api_def_map TF_GUARDED_BY(lock); #endif // !defined(IS_MOBILE_PLATFORM) && !defined(IS_SLIM_BUILD) bool update_docs_called TF_GUARDED_BY(lock);
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Sat May 13 00:49:12 GMT 2023 - 7.6K bytes - Viewed (0) -
tensorflow/c/eager/immediate_execution_tensor_handle.h
// ImmediateExecutionTensorHandles, but comes up for example when computing // the shape of a parallel tensor with component shapes differing across // devices. virtual Status Dim(int dim_index, int64_t* dim) const = 0; // Returns the device which created the handle. virtual const char* DeviceName(Status* status) const = 0; // Returns the device where the tensor was placed.
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Mar 10 21:56:24 GMT 2023 - 4.3K bytes - Viewed (0) -
tensorflow/c/c_api_test.cc
#undef EXPECT_TF_META TEST(CAPI, TestTensorAligned) { int64_t dim = 7; size_t tensor_size_bytes = dim * TF_DataTypeSize(TF_FLOAT); TF_Tensor* a = TF_AllocateTensor( /*dtype=*/TF_FLOAT, /*dims=*/&dim, /*num_dims=*/1, /*len=*/tensor_size_bytes); float* data = reinterpret_cast<float*>(TF_TensorData(a)); for (int i = 0; i < dim; ++i) { data[i] = 0; } if (TF_TensorDefaultAlignment() > 0) {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 96.9K bytes - Viewed (3) -
tensorflow/c/eager/c_api_test.cc
ASSERT_EQ(num_dims, -1); ASSERT_EQ("Invalid handle", string(TF_Message(status.get()))); TF_SetStatus(status.get(), TF_OK, ""); int dim = TFE_TensorHandleDim(h, 0, status.get()); ASSERT_EQ(TF_INVALID_ARGUMENT, TF_GetCode(status.get())); ASSERT_EQ(dim, -1); ASSERT_EQ("Invalid handle", string(TF_Message(status.get()))); } TEST(CAPI, TensorHandleDevices) {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Aug 03 20:50:20 GMT 2023 - 94.6K bytes - Viewed (1) -
tensorflow/c/c_api.cc
#if !defined(IS_MOBILE_PLATFORM) && !defined(IS_SLIM_BUILD) delete server; #endif // !defined(IS_MOBILE_PLATFORM) && !defined(IS_SLIM_BUILD) } void TF_RegisterLogListener(void (*listener)(const char*)) { #if !defined(IS_MOBILE_PLATFORM) && !defined(IS_SLIM_BUILD) tensorflow::logging::RegisterListener(listener); #endif // !defined(IS_MOBILE_PLATFORM) && !defined(IS_SLIM_BUILD) }
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.cc
auto* tf_dlm_tensor_ctx = new TfDlManagedTensorCtx(tensor_ref); tf_dlm_tensor_ctx->reference = tensor_ref; DLManagedTensor* dlm_tensor = &tf_dlm_tensor_ctx->tensor; dlm_tensor->manager_ctx = tf_dlm_tensor_ctx; dlm_tensor->deleter = &DLManagedTensorDeleter; dlm_tensor->dl_tensor.device = tf_dlm_context; int ndim = tensor->dims(); dlm_tensor->dl_tensor.ndim = ndim; dlm_tensor->dl_tensor.data = tf_dlm_data;
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/eager/dlpack_test.cc
data[j] = j; } DLManagedTensor dlm_in = {}; DLTensor* dltensor_in = &dlm_in.dl_tensor; dltensor_in->data = data.data(); dltensor_in->device = {kDLCPU, 0}; dltensor_in->ndim = static_cast<int32_t>(shape.size()); dltensor_in->dtype = {kDLFloat, 32, 1}; dltensor_in->shape = shape.data(); dltensor_in->strides = strides.data(); TFE_TensorHandle* handle = TFE_HandleFromDLPack(&dlm_in, status, ctx); ASSERT_NE(handle, nullptr)
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/gradients/nn_grad.cc
} auto imm_ctx = dyn_cast<ImmediateExecutionContext>(ctx); AbstractTensorPtr minus_1(imm_ctx->CreateInt32Scalar(-1)); ImmediateTensorHandlePtr dim(imm_ctx->CreateLocalHandle(minus_1.get())); AbstractTensorHandle* expand_dims_outputs; TF_RETURN_IF_ERROR( ops::ExpandDims(ctx, vec, dim.get(), &expand_dims_outputs, "ExpandDims")); TF_RETURN_IF_ERROR( ops::Mul(ctx, expand_dims_outputs, mat, &outputs[0], "Mul"));
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 5.7K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental.cc
return nullptr; } 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; }
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/eager/c_api.h
TF_CAPI_EXPORT extern int TFE_TensorDebugInfoOnDeviceNumDims( TFE_TensorDebugInfo* debug_info); // Returns the number of elements in dimension `dim_index`. // Tensor representation on device can be transposed from its representation // on host. The data contained in dimension `dim_index` on device // can correspond to the data contained in another dimension in on-host // representation. The dimensions are indexed using the standard TensorFlow
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Apr 27 21:07:00 GMT 2023 - 22.8K bytes - Viewed (1)