Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 9 of 9 for Builder (0.17 sec)

  1. tensorflow/c/eager/c_api_unified_experimental_graph.cc

        tensorflow::StringPiece s(data, length);
        op_->node_builder.Attr(attr_name, s);
        return absl::OkStatus();
      }
      Status SetAttrInt(const char* attr_name, int64_t value) override {
        op_->node_builder.Attr(attr_name, static_cast<int64_t>(value));
        return absl::OkStatus();
      }
      Status SetAttrFloat(const char* attr_name, float value) override {
        op_->node_builder.Attr(attr_name, value);
        return absl::OkStatus();
      }
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 15.4K bytes
    - Viewed (1)
  2. tensorflow/c/c_api.cc

      Node* ret = nullptr;
    
      if (desc->graph->name_map.count(desc->node_builder.node_name())) {
        status->status = InvalidArgument("Duplicate node name in graph: '",
                                         desc->node_builder.node_name(), "'");
      } else {
        if (!desc->colocation_constraints.empty()) {
          desc->node_builder.Attr(
              tensorflow::kColocationAttrName,
    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)
  3. tensorflow/c/c_api_experimental.cc

    }
    
    void TF_DeleteAttrBuilder(TF_AttrBuilder* builder) { delete builder; }
    
    void TF_AttrBuilderSetType(TF_AttrBuilder* builder, const char* attr_name,
                               TF_DataType value) {
      auto iter = builder->attr_names.insert(attr_name).first;
      builder->Set(*iter, static_cast<tensorflow::DataType>(value));
    }
    
    void TF_AttrBuilderSetTypeList(TF_AttrBuilder* builder, const char* attr_name,
    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)
  4. tensorflow/c/c_api_test.cc

    };
    
    // Test we can query kernels
    REGISTER_OP("TestOpWithSingleKernel")
        .Input("a: float")
        .Input("b: float")
        .Output("o: float");
    REGISTER_KERNEL_BUILDER(
        Name("TestOpWithSingleKernel").Device(tensorflow::DEVICE_CPU), DummyKernel);
    
    TEST(TestKernel, TestGetAllRegisteredKernels) {
      TF_Status* status = TF_NewStatus();
    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)
  5. tensorflow/c/eager/BUILD

                "//tensorflow/core:lib",
                "//tensorflow/core:lib_internal",
                "//tensorflow/core:protos_all_cc",
                "//tensorflow/core/common_runtime/eager:attr_builder",
                "//tensorflow/core/common_runtime/eager:context",
                "//tensorflow/core/common_runtime/eager:context_distributed_manager",
                "//tensorflow/core/common_runtime/eager:core",
    Plain Text
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Apr 11 23:52:39 GMT 2024
    - 33.3K bytes
    - Viewed (0)
  6. tensorflow/BUILD

    #     visibility = ["//visibility:public"],
    # )
    # copybara:uncomment_end
    
    # 'enable_registration_v2' opts-in to a different implementation of op and
    # kernel registration - REGISTER_OP, REGISTER_KERNEL_BUILDER, etc.
    #
    # This setting is currently experimental. The 'v2' implementation does _not_
    # correspond to a particular, finalized design; rather, it relates to
    # developing one.
    #
    Plain Text
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Apr 09 18:15:11 GMT 2024
    - 53.4K bytes
    - Viewed (8)
  7. tensorflow/c/BUILD

            "//tensorflow/core:core_cpu",
            "//tensorflow/core:framework",
            "//tensorflow/core:lib",
            "//tensorflow/core:protos_all_cc",
            "//tensorflow/core/common_runtime/eager:attr_builder",
            "//tensorflow/core/common_runtime/eager:context",
            "//tensorflow/core/common_runtime/eager:core",
            "//tensorflow/core/common_runtime/eager:eager_operation",
    Plain Text
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Mar 27 18:00:18 GMT 2024
    - 30.3K bytes
    - Viewed (0)
  8. tensorflow/c/eager/c_api.cc

    #include "tensorflow/core/common_runtime/device.h"
    #include "tensorflow/core/common_runtime/device_factory.h"
    #include "tensorflow/core/common_runtime/device_mgr.h"
    #include "tensorflow/core/common_runtime/eager/attr_builder.h"
    #include "tensorflow/core/common_runtime/eager/context.h"
    #include "tensorflow/core/common_runtime/eager/custom_device.h"
    #include "tensorflow/core/common_runtime/eager/custom_device_op_handler.h"
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 43.9K bytes
    - Viewed (2)
  9. RELEASE.md

    Karnewar, Ankit Gupta, Anton Matosov, Aris L, Ben Barsdell, Brent Yi, Brett
    Koonce, Carl Thomé, cbockman, Chikanaga Tomoyuki, Chris Tava, CéDric Deltheil,
    Dahan Gong, Dalmo Cirne, Daniel Erenrich, David Norman, DavidNorman, Edd
    Wilder-James, Fanjin Zeng, Felix Abecassis, fo40225, George Sterpu, Giovanni
    Terlingen, Gor Baghdasaryan, Guillaume Klein, Hanchen Li, Ilya Polenov, Jakub
    Kolodziejczyk, Jason Sadler, Jayaram Bobba, Jerry Liu, jinghuangintel, Jiongyan
    Plain Text
    - Registered: Tue May 07 12:40:20 GMT 2024
    - Last Modified: Mon Apr 29 19:17:57 GMT 2024
    - 727.7K bytes
    - Viewed (8)
Back to top