Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for Operator (0.21 sec)

  1. tensorflow/c/c_api.cc

      return oper->node.attrs().size();
    }
    
    int TF_OperationGetAttrNameLength(TF_Operation* oper, int i) {
      auto attrs = oper->node.attrs();
      int count = 0;
      AttrValueMap::const_iterator it;
      for (it = attrs.begin(); it != attrs.end(); it++) {
        if (count == i) {
          return it->first.length();
        }
        count++;
      }
      return -1;
    }
    
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 102.3K bytes
    - Viewed (0)
  2. tensorflow/c/eager/parallel_device/parallel_device_lib.cc

    class OpDeleter {
     public:
      void operator()(TFE_Op* to_delete) const { TFE_DeleteOp(to_delete); }
    };
    
    using OpPtr = std::unique_ptr<TFE_Op, OpDeleter>;
    
    class StatusDeleter {
     public:
      void operator()(TF_Status* to_delete) const { TF_DeleteStatus(to_delete); }
    };
    
    using StatusPtr = std::unique_ptr<TF_Status, StatusDeleter>;
    
    class ExecutorDeleter {
     public:
      void operator()(TFE_Executor* to_delete) const {
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Fri Feb 09 07:47:20 GMT 2024
    - 25.4K bytes
    - Viewed (1)
  3. tensorflow/c/eager/c_api_test.cc

      TFE_OpAddInput(minOp, axis, status);
      CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
    
      tensorflow::AttrValueMap attr_values = ExtractAttrs(minOp);
      tensorflow::AttrValueMap::const_iterator attr_found = attr_values.find("T");
      EXPECT_NE(attr_found, attr_values.cend());
      EXPECT_EQ(attr_found->second.type(), tensorflow::DataType::DT_FLOAT);
      attr_found = attr_values.find("Tidx");
    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)
  4. tensorflow/c/c_api_test.cc

    limitations under the License.
    ==============================================================================*/
    
    #include "tensorflow/c/c_api.h"
    
    #include <algorithm>
    #include <cstddef>
    #include <iterator>
    #include <memory>
    #include <vector>
    
    #include "tensorflow/c/c_api_internal.h"
    #include "tensorflow/c/c_test_util.h"
    #include "tensorflow/c/tf_buffer.h"
    #include "tensorflow/c/tf_buffer_internal.h"
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 96.9K bytes
    - Viewed (3)
  5. tensorflow/c/eager/parallel_device/parallel_device.cc

    #include "tensorflow/c/tf_status.h"
    #include "tensorflow/c/tf_status_helper.h"
    
    namespace tensorflow {
    namespace parallel_device {
    namespace {
    
    class OpDeleter {
     public:
      void operator()(TFE_Op* to_delete) const { TFE_DeleteOp(to_delete); }
    };
    
    using OpPtr = std::unique_ptr<TFE_Op, OpDeleter>;
    
    using MaybeParallelTensorOwned =
        absl::variant<std::unique_ptr<ParallelTensor>, TensorHandlePtr>;
    
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Wed Mar 29 22:05:31 GMT 2023
    - 18.3K bytes
    - Viewed (0)
Back to top