Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 7 of 7 for TFE_OpSetAttrString (0.23 sec)

  1. tensorflow/c/eager/c_api_test_util.cc

      TF_DeleteStatus(status);
    
      TFE_OpSetAttrType(op, "T", TFE_TensorHandleDataType(in));
      TFE_OpSetAttrString(op, "tensor_name", "dummy", 5);
      TFE_OpSetAttrString(op, "send_device", send_device.c_str(),
                          send_device.size());
      TFE_OpSetAttrString(op, "recv_device", recv_device.c_str(),
                          recv_device.size());
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Wed Feb 21 22:37:46 GMT 2024
    - 23.5K bytes
    - Viewed (2)
  2. tensorflow/c/eager/custom_device_test.cc

      ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get());
      TFE_OpSetAttrType(op.get(), "dtype", TF_FLOAT);
      TFE_OpSetAttrShape(op.get(), "shape", {}, 0, status.get());
      TFE_OpSetAttrString(op.get(), "container", "", 0);
      TFE_OpSetAttrString(op.get(), "shared_name", "", 0);
      TFE_OpSetDevice(op.get(), name, status.get());
      ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get());
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Thu Aug 27 23:39:24 GMT 2020
    - 18.4K bytes
    - Viewed (0)
  3. tensorflow/c/eager/parallel_device/parallel_device_testlib.cc

      TFE_OpSetAttrString(op.get(), "container", "", 0);
      // Use the special GUID for no buffer sharing
      //
      // TODO(allenl): Should we provide a better API for this? AFAIK this is the
      // only reasonable way to make variables with no aliasing using the eager C
      // API.
      std::string no_sharing = "cd2c89b7-88b7-44c8-ad83-06c2a9158347";
      TFE_OpSetAttrString(op.get(), "shared_name", no_sharing.c_str(),
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Tue Jun 15 15:44:44 GMT 2021
    - 12.5K bytes
    - Viewed (0)
  4. tensorflow/c/eager/parallel_device/parallel_device_test.cc

      TFE_OpSetAttrInt(op.get(), "group_key", 0);
      TFE_OpSetAttrInt(op.get(), "instance_key", 0);
      const std::string merge_op("Add");
      TFE_OpSetAttrString(op.get(), "merge_op", merge_op.c_str(),
                          merge_op.length());
      const std::string final_op("Id");
      TFE_OpSetAttrString(op.get(), "final_op", final_op.c_str(),
                          final_op.length());
      TFE_OpSetAttrIntList(op.get(), "subdiv_offsets", nullptr, 0);
    
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Thu Jul 08 23:47:35 GMT 2021
    - 29.3K bytes
    - Viewed (1)
  5. tensorflow/c/eager/c_api_test.cc

      char buffer[BUFFER_SIZE];
      std::strncpy(buffer, "VALID", BUFFER_SIZE);
      TFE_OpSetAttrString(op, "padding", buffer, std::strlen(buffer));
      // Overwriting value in "buffer", should be fine since TFE_Op
      // shouldn't be holding on to it.
      std::strncpy(buffer, "NHWC", BUFFER_SIZE);
      TFE_OpSetAttrString(op, "data_format", buffer, std::strlen(buffer));
    
      TFE_OpSetAttrType(op, "T", TF_FLOAT);
    
    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)
  6. tensorflow/c/eager/c_api.cc

        ret = TFE_OpGetAttrType(op, attr_name, is_list, status);
      } else {
        ret = TF_ATTR_INT;  // Same dummy return as TFE_OpGetAttrType.
      }
      TFE_DeleteOp(op);
      return ret;
    }
    
    void TFE_OpSetAttrString(TFE_Op* op, const char* attr_name, const void* value,
                             size_t length) {
      auto s = tensorflow::unwrap(op)->SetAttrString(
          attr_name, static_cast<const char*>(value), length);
    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)
  7. tensorflow/c/eager/c_api.h

    TF_CAPI_EXPORT extern TF_AttrType TFE_OpNameGetAttrType(
        TFE_Context* ctx, const char* op_or_function_name, const char* attr_name,
        unsigned char* is_list, TF_Status* status);
    
    TF_CAPI_EXPORT extern void TFE_OpSetAttrString(TFE_Op* op,
                                                   const char* attr_name,
                                                   const void* value,
                                                   size_t length);
    C
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Thu Apr 27 21:07:00 GMT 2023
    - 22.8K bytes
    - Viewed (1)
Back to top