Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 7 of 7 for Builder (0.2 sec)

  1. 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)
  2. 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)
  3. tensorflow/c/eager/c_api_test.cc

      void Compute(tensorflow::OpKernelContext* ctx) override {
        ctx->SetStatus(tensorflow::errors::Unavailable("Test error."));
      }
    };
    REGISTER_KERNEL_BUILDER(
        Name("TestNonCommUnavailable").Device(tensorflow::DEVICE_DEFAULT),
        TestUnavailableErrorOp);
    REGISTER_KERNEL_BUILDER(
        Name("TestCommUnavailable").Device(tensorflow::DEVICE_DEFAULT),
        TestUnavailableErrorOp);
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Aug 03 20:50:20 GMT 2023
    - 94.6K bytes
    - Viewed (1)
  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/gradients.cc

    #include "tensorflow/c/eager/abstract_tensor_handle.h"
    #include "tensorflow/c/eager/c_api_unified_experimental_internal.h"
    #include "tensorflow/c/eager/gradients_internal.h"
    #include "tensorflow/core/common_runtime/eager/attr_builder.h"
    #include "tensorflow/core/lib/llvm_rtti/llvm_rtti.h"
    #include "tensorflow/core/platform/errors.h"
    
    namespace tensorflow {
    namespace gradients {
    namespace {
    
    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)
  6. 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)
  7. 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)
Back to top