Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for SetOpAttrValueScalar (0.19 sec)

  1. tensorflow/c/eager/tfe_op_attrs_internal.h

    typedef struct TFE_Op TFE_Op;
    
    namespace tensorflow {
    DEFINE_CONVERSION_FUNCTIONS(tensorflow::AbstractOpAttrs, TFE_OpAttrs);
    
    // Set an AttrValue on the op. Doesn't handle the list types.
    void SetOpAttrValueScalar(TFE_Context* ctx, TFE_Op* op,
                              const tensorflow::AttrValue& default_value,
                              const char* attr_name, TF_Status* status);
    }  // namespace tensorflow
    
    C
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Wed Feb 10 05:41:19 GMT 2021
    - 1.6K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api_test.cc

      for (int i = 0; i < 4; ++i) {
        i_list_values.mutable_list()->add_i(1);
      }
      SetOpAttrValueScalar(ctx, op, i_list_values, "ksize", status);
      SetOpAttrValueScalar(ctx, op, i_list_values, "strides", status);
    
      tensorflow::AttrValue padding_value;
      *padding_value.mutable_s() = "VALID";
      tensorflow::SetOpAttrValueScalar(ctx, op, padding_value, "padding", status);
    
      tensorflow::AttrValue data_format_value;
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Thu Aug 03 20:50:20 GMT 2023
    - 94.6K bytes
    - Viewed (1)
  3. tensorflow/c/eager/c_api.cc

                    TF_Status* status) {
      TFE_Op* func_op = TFE_NewOp(ctx, func.name().data(), status);
      for (const auto& attr : func.attr()) {
        if (!status->status.ok()) return nullptr;
        SetOpAttrValueScalar(ctx, func_op, attr.second, attr.first.data(), status);
        if (!status->status.ok()) return nullptr;
      }
      return func_op;
    }
    }  // namespace
    
    void TFE_ContextStartStep(TFE_Context* ctx) {
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 43.9K bytes
    - Viewed (2)
Back to top