- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 13 for Garg (0.17 sec)
-
tensorflow/c/experimental/next_pluggable_device/c_api.cc
TF_Status* status) { auto* cc_ctx = reinterpret_cast<tensorflow::OpKernelContext*>(ctx); const tensorflow::Tensor& arg_tensor = cc_ctx->input(index); absl::Status cc_status; if (arg_tensor.dtype() != tensorflow::DT_RESOURCE) { cc_status = absl::InvalidArgumentError( absl::StrCat("Trying to obtain resource handle from Input[", index,
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/c_api_distributed_test.cc
CHECK(tensorflow::protobuf::TextFormat::ParseFromString( " signature {" " name: 'AddVariablesFunction'" " input_arg {" " name: 'var'" " type: DT_RESOURCE" " }" " output_arg {" " name: 'sum'" " type: DT_FLOAT" " }" " }" " node_def {" " name: 'read0'"
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 23.5K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_lib.h
TF_NewTensor( datatype_enum, /*dims=*/nullptr, /*num_dims=*/0, device_value.release(), sizeof(DataType), [](void* data, size_t, void* arg) { delete reinterpret_cast<DataType*>(data); }, nullptr), TF_DeleteTensor); // TODO(allenl): Here and when executing regular operations, we could hold
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Apr 25 15:21:13 GMT 2023 - 12.9K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_testlib.cc
GeneralAssignment("AssignVariableOp", context, value, status); } // Passed to `TF_NewTensor` to indicate how an array of floats should be // deleted. static void FloatDeallocator(void* data, size_t, void* arg) { delete[] static_cast<float*>(data); } // Creates a TFE_TensorHandle with value `v`. TensorHandlePtr FloatTensorHandle(float v, TF_Status* status) { const int num_bytes = sizeof(float);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Jun 15 15:44:44 GMT 2021 - 12.5K bytes - Viewed (0) -
tensorflow/c/eager/c_api_experimental_test.cc
// First create a simple identity function. TF_Graph* function_graph = TF_NewGraph(); TF_OperationDescription* arg_descr = TF_NewOperation(function_graph, "Placeholder", "arg"); TF_SetAttrType(arg_descr, "dtype", TF_INT32); TF_Status* status = TF_NewStatus(); TF_Operation* arg = TF_FinishOperation(arg_descr, status); ASSERT_TRUE(TF_GetCode(status) == TF_OK) << TF_Message(status); TF_OperationDescription* id_descr =
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Aug 03 03:14:26 GMT 2023 - 31.5K bytes - Viewed (1) -
tensorflow/c/eager/c_api_unified_experimental_test.cc
// Build two abstract input tensors as function arguments. std::vector<TF_AbstractTensor*> func_args; { TFE_Context* eager_ctx = TF_ExecutionContextGetTFEContext(eager_execution_ctx, s); // 1st Arg float vals1[] = {0.0f, 1.0f, 1.0f, 0.0f}; int64_t dims[] = {2, 2}; // Matrices will be 2 x 2 int num_dims = sizeof(dims) / sizeof(dims[0]); TFE_TensorHandle* input_eager =
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri May 19 21:44:52 GMT 2023 - 39.1K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.cc
if (!status->status.ok()) return nullptr; if (input_index >= op_def->input_arg_size() || input_index < 0) { status->status = tensorflow::errors::InvalidArgument( input_index, " out of range for ", op_name); return nullptr; } const tensorflow::OpDef_ArgDef& input_arg = op_def->input_arg()[input_index]; if (input_arg.number_attr().empty()) { status->status = tensorflow::errors::NotFound(
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/c_api_experimental.h
// `tensor_id`, associated with `session`. There must be a graph node named // "fifo_queue_enqueue_<tensor_id>", to be executed by this API call. It reads // from a placeholder node "arg_tensor_enqueue_<tensor_id>". // // `tensor` is still owned by the caller. This call will be blocked if the queue // has reached its capacity, and will be unblocked when the queued tensors again
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Apr 27 21:07:00 GMT 2023 - 15.1K bytes - Viewed (0) -
tensorflow/c/c_test_util.cc
static void BoolDeallocator(void* data, size_t, void* arg) { delete[] static_cast<bool*>(data); } static void Int32Deallocator(void* data, size_t, void* arg) { delete[] static_cast<int32_t*>(data); } static void DoubleDeallocator(void* data, size_t, void* arg) { delete[] static_cast<double*>(data); } static void FloatDeallocator(void* data, size_t, void* arg) { delete[] static_cast<float*>(data); }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Oct 15 03:16:52 GMT 2021 - 17.8K bytes - Viewed (2) -
tensorflow/c/eager/dlpack.cc
if (!status->status.ok()) { return nullptr; } return tensor; } // Deleter for DLManagedTensor void DLManagedTensorDeleter(DLManagedTensor* arg) { TfDlManagedTensorCtx* owner = static_cast<TfDlManagedTensorCtx*>(arg->manager_ctx); owner->reference.Unref(); delete owner; } // Converts TF_DATAType to DLPack data type. DLDataType GetDlDataType(TF_DataType data_type, TF_Status* status) {
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)