Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for BuildNodeDef (0.16 sec)

  1. 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)
  2. 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)
  3. 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)
Back to top