- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for BuildNodeDef (0.16 sec)
-
tensorflow/c/experimental/gradients/tape/tape_operation.cc
// attributes. Number type attrs and DataType attrs work fine without this. // Consider getting rid of this and making the behavior between number types // and string consistent. forward_op_.attrs.BuildNodeDef(); // TODO(b/170307493): Populate skip_input_indices here. std::unique_ptr<GradientFunction> backward_fn; TF_RETURN_IF_ERROR(registry_.Lookup(forward_op_, &backward_fn));
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Tue Jun 07 01:53:35 GMT 2022 - 9K bytes - Viewed (1) -
tensorflow/c/eager/gradients.cc
// attributes. Number type attrs and DataType attrs work fine without this. // Consider getting rid of this and making the behavior between number types // and string consistent. forward_op_->attrs.BuildNodeDef(); std::unique_ptr<GradientFunction> gradient_fn; TF_RETURN_IF_ERROR(registry.Lookup(*forward_op_, &gradient_fn)); tape->RecordOperation(forward_op_->inputs, retvals, gradient_fn.release(),
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/c_api_experimental.cc
const char* device_type, TF_Status* status) { status->status = tensorflow::FindKernelDef( tensorflow::DeviceType(device_type), builder->BuildNodeDef(), /* def = */ nullptr, /* kernel_class_name = */ nullptr); } const char* TF_GetNumberAttrForOpListInput(const char* op_name, int input_index, TF_Status* status) {
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)