- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 25 for size_op (0.09 sec)
-
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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)