Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 7 of 7 for tf_attrtype (0.1 sec)

  1. tensorflow/c/BUILD

            ],
            ("//tensorflow:chromiumos", "//tensorflow:fuchsia"): [
                ":tf_attrtype",
                "//tensorflow/core:core_cpu",
                "//tensorflow/core:framework",
                "//tensorflow/core:lib",
                "//tensorflow/core/platform:platform",
            ],
            "//conditions:default": [
                ":tf_attrtype",
                "//tensorflow/core:core_cpu",
                "//tensorflow/core:framework",
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Nov 02 06:47:06 UTC 2024
    - 30.4K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api.cc

                                                TF_Status* status) {
      return tensorflow::wrap(tensorflow::unwrap(op)->GetInputs()[index]);
    }
    
    TF_AttrType TFE_OpGetAttrType(TFE_Op* op, const char* attr_name,
                                  unsigned char* is_list, TF_Status* status) {
      TF_AttrType ret = TF_ATTR_INT;
      const tensorflow::AttrTypeMap* attr_types_;
      bool is_function;
      status->status = tensorflow::AttrTypeMapForOp(
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 43.9K bytes
    - Viewed (0)
  3. tensorflow/c/eager/c_api.h

                                                               int index,
                                                               TF_Status* status);
    
    TF_CAPI_EXPORT extern TF_AttrType TFE_OpGetAttrType(TFE_Op* op,
                                                        const char* attr_name,
                                                        unsigned char* is_list,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Apr 27 21:07:00 UTC 2023
    - 22.8K bytes
    - Viewed (0)
  4. tensorflow/c/c_api.h

    #ifndef TENSORFLOW_C_C_API_H_
    #define TENSORFLOW_C_C_API_H_
    
    #include <stddef.h>
    #include <stdint.h>
    
    #include "tensorflow/c/c_api_macros.h"
    #include "tensorflow/c/tf_attrtype.h"
    #include "tensorflow/c/tf_buffer.h"
    #include "tensorflow/c/tf_datatype.h"
    #include "tensorflow/c/tf_status.h"
    #include "tensorflow/c/tf_tensor.h"
    #include "tensorflow/c/tf_tstring.h"
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Oct 26 21:08:15 UTC 2023
    - 82.3K bytes
    - Viewed (0)
  5. tensorflow/c/c_api.cc

        SINGLE_CASE(kS, TF_ATTR_STRING, attr->s().length());
        SINGLE_CASE(kI, TF_ATTR_INT, -1);
        SINGLE_CASE(kF, TF_ATTR_FLOAT, -1);
        SINGLE_CASE(kB, TF_ATTR_BOOL, -1);
        SINGLE_CASE(kType, TF_ATTR_TYPE, -1);
        SINGLE_CASE(kShape, TF_ATTR_SHAPE,
                    attr->shape().unknown_rank() ? -1 : attr->shape().dim_size());
        SINGLE_CASE(kTensor, TF_ATTR_TENSOR, -1);
    #undef SINGLE_CASE
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 102.3K bytes
    - Viewed (0)
  6. tensorflow/c/c_api_test.cc

      auto desc = init("type");
      TF_SetAttrType(desc, "v", TF_COMPLEX128);
    
      auto oper = TF_FinishOperation(desc, s_);
      ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_);
      EXPECT_TF_META("v", -1, TF_ATTR_TYPE, -1);
    
      TF_DataType value;
      TF_OperationGetAttrType(oper, "v", &value, s_);
      EXPECT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_);
      EXPECT_EQ(TF_COMPLEX128, value);
    }
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 97K bytes
    - Viewed (0)
  7. tensorflow/c/eager/c_api_test.cc

      TFE_Op* copy_op = TFE_NewOp(ctx, "VarHandleOp", status);
      TFE_OpSetAttrType(copy_op, "dtype", TF_FLOAT);
      TFE_OpAddAttrs(copy_op, attributes);
      unsigned char is_list = 0;
      ASSERT_EQ(TF_ATTR_TYPE,
                TFE_OpGetAttrType(copy_op, "dtype", &is_list, status));
      CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
      ASSERT_EQ(TF_ATTR_SHAPE,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Aug 03 20:50:20 UTC 2023
    - 94.6K bytes
    - Viewed (0)
Back to top