Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 25 for size_op (0.07 sec)

  1. tensorflow/c/eager/parallel_device/parallel_device_lib_test.cc

      EXPECT_THAT(*shape, ElementsAre(-1));
      std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> size_op(
          TFE_NewOp(context.get(), "Size", status.get()), TFE_DeleteOp);
      auto result = parallel_device.Execute(
          context.get(), {unknown_dims_vector.get()}, "Size",
          TFE_OpGetAttrs(size_op.get()), 1, status.get());
      ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK);
      TF_ASSERT_OK((*result)[0]->Shape(&shape));
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Mon Oct 21 04:14:14 UTC 2024
    - 15.6K bytes
    - Viewed (0)
  2. tensorflow/c/c_api_test.cc

    void StringVectorToArrays(const std::vector<string>& v,
                              std::unique_ptr<const void*[]>* ptrs,
                              std::unique_ptr<size_t[]>* lens) {
      ptrs->reset(new const void*[v.size()]);
      lens->reset(new size_t[v.size()]);
      for (size_t i = 0; i < v.size(); ++i) {
        (*ptrs)[i] = v[i].data();
        (*lens)[i] = v[i].size();
      }
    }
    
    class CApiColocationTest : public ::testing::Test {
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 97K bytes
    - Viewed (0)
  3. tensorflow/c/eager/parallel_device/parallel_device_lib.h

          const ParallelDevice& parallel_device,
          std::vector<TensorHandlePtr> components, absl::Span<const int64_t> shape,
          TF_Status* status);
    
      size_t num_tensors() const { return tensors_.size(); }
      TFE_TensorHandle* tensor(size_t index) const { return tensors_[index].get(); }
    
      // If the `shape` argument to `FromTensorHandles` is specified, returns that.
      //
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Mon Oct 21 04:14:14 UTC 2024
    - 13.1K bytes
    - Viewed (0)
  4. tensorflow/c/c_api_experimental.cc

    }
    
    const char* TF_GraphDebugString(TF_Graph* graph, size_t* len) {
      tensorflow::mutex_lock c(graph->mu);
      const auto& debug_str = graph->graph.ToGraphDefDebug().DebugString();
      *len = debug_str.size();
      char* ret = static_cast<char*>(malloc(*len + 1));
      memcpy(ret, debug_str.c_str(), *len + 1);
      return ret;
    }
    
    char* TF_FunctionDebugString(TF_Function* func, size_t* len) {
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 29.5K bytes
    - Viewed (0)
  5. tensorflow/c/eager/c_api.cc

          methods));
    }
    
    TFE_TensorHandle* TFE_NewTensorHandleFromDeviceMemory(
        TFE_Context* ctx, const char* device_name, TF_DataType dtype,
        const int64_t* dims, int num_dims, void* data, size_t len,
        void (*deallocator)(void* data, size_t len, void* arg),
        void* deallocator_arg, TF_Status* status) {
      tensorflow::Device* device = nullptr;
      tensorflow::EagerContext* context =
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 43.9K bytes
    - Viewed (0)
  6. tensorflow/c/eager/abstract_operation.h

                                               size_t length) = 0;
      virtual absl::Status SetAttrTensor(const char* attr_name,
                                         AbstractTensorInterface* tensor) = 0;
      virtual absl::Status SetAttrStringList(const char* attr_name,
                                             const void* const* values,
                                             const size_t* lengths,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 7.3K bytes
    - Viewed (0)
  7. tensorflow/c/eager/c_api_experimental.cc

      void* data = tensorflow::port::Malloc(value.length());
      value.copy(static_cast<char*>(data), value.length(), 0);
      buf->data = data;
      buf->length = value.length();
      buf->data_deallocator = [](void* data, size_t length) {
        tensorflow::port::Free(data);
      };
    }
    
    TFE_MonitoringStringGauge0* TFE_MonitoringNewStringGauge0(
        const char* name, TF_Status* status, const char* description) {
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 35.9K bytes
    - Viewed (0)
  8. tensorflow/c/c_api.cc

      }
      const auto& s = attr->s();
      std::memcpy(value, s.data(), std::min<size_t>(s.length(), max_length));
    }
    
    void TF_OperationGetAttrStringList(TF_Operation* oper, const char* attr_name,
                                       void** values, size_t* lengths,
                                       int max_values, void* storage,
                                       size_t storage_size, TF_Status* status) {
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 102.3K bytes
    - Viewed (0)
  9. tensorflow/c/eager/c_api_unified_experimental_test.cc

        [[0,0],    *   [[0,0],    =   [[0,0],
         [0,0]]         [0,0]]         [0,0]]
      */
    
      // Build an abstract input tensor.
      int64_t dims[] = {2, 2};  // Matrices will be 2 x 2
      int num_dims = sizeof(dims) / sizeof(dims[0]);
    
      float vals[] = {0.0f, 0.0f, 0.0f, 0.0f};
      TFE_Context* eager_ctx = TF_ExecutionContextGetTFEContext(ctx, status.get());
      TFE_TensorHandle* t =
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 39.1K bytes
    - Viewed (0)
  10. tensorflow/c/eager/gradients_internal.h

                                     const char* value, size_t length,
                                     ForwardOperation*);
    absl::Status SetAttrTensor(AbstractOperation*, const char* attr_name,
                               AbstractTensorInterface* tensor, ForwardOperation*);
    absl::Status SetAttrStringList(AbstractOperation*, const char* attr_name,
                                   const void* const* values, const size_t* lengths,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 4.7K bytes
    - Viewed (0)
Back to top