- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 44 for Kata (0.14 sec)
-
tensorflow/c/eager/dlpack_test.cc
dltensor_out->shape[i + 1] * dltensor_out->strides[i + 1]); } } } const float* data_in = static_cast<const float*>(dltensor_in->data); const float* data_out = static_cast<const float*>(dltensor_out->data); for (size_t j = 0; j < num_elements; ++j) { EXPECT_EQ(data_out[j], data_in[j]); } TFE_CallDLManagedTensorDeleter(dlm_out); TFE_DeleteTensorHandle(handle); }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Jun 30 03:04:46 GMT 2023 - 4.4K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/nn_grad.cc
*/ AbstractTensorHandle* upstream_grad = grad_outputs[0]; DCHECK(upstream_grad); // Recover data format from forward pass for gradient. std::string data_format; TF_RETURN_IF_ERROR(forward_attrs_.Get("data_format", &data_format)); // Grad for A grad_inputs[0] = upstream_grad; grad_inputs[0]->Ref(); // Grad for bias
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 5.7K bytes - Viewed (0) -
tensorflow/c/c_api.cc
auto* out_shape_and_type = handle_data.add_shape_and_type(); ic->ShapeHandleToProto(p.shape, out_shape_and_type->mutable_shape()); out_shape_and_type->set_dtype(p.dtype); *out_shape_and_type->mutable_type() = p.type; } } string str_data; handle_data.SerializeToString(&str_data); TF_Buffer* result = TF_NewBufferFromString(str_data.c_str(), str_data.size()); return result; }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 102.3K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_remote_test.cc
server_def, tensorflow::Env::Default(), &worker_server2) .ok()); ASSERT_TRUE(worker_server2->Start().ok()); TFE_ContextSetServerDef(context.get(), 0, serialized.data(), serialized.size(), status.get()); EXPECT_EQ(TF_OK, TF_GetCode(status.get())) << TF_Message(status.get()); BasicTestsForTwoDevices(context.get(),
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Apr 27 22:09:57 GMT 2023 - 6.7K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental.cc
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 9K bytes - Viewed (0) -
tensorflow/c/eager/c_api_cluster_test.cc
TFE_ContextUpdateServerDef(ctx, 0, serialized_update.data(), serialized_update.size(), status); EXPECT_NE(TF_OK, TF_GetCode(status)) << TF_Message(status); // Even after the prevoiusly failed cluster update, another update and op // execution should work fine as long as the provided server_def is valid. TFE_ContextUpdateServerDef(ctx, 0, serialized.data(), serialized.size(),
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Apr 14 10:03:59 GMT 2023 - 19.3K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.cc
} int TF_OpIsStateful(const char* op_type, TF_Status* status) { const tensorflow::OpRegistrationData* op_reg_data; status->status = tensorflow::OpRegistry::Global()->LookUp(op_type, &op_reg_data); if (!status->status.ok()) { return 0; } return op_reg_data->op_def.is_stateful(); } void TF_InitMain(const char* usage, int* argc, char*** argv) {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 29.4K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_lib_test.cc
TF_CreateConfig( /*xla*/ false, /* gpu_memory_allow_growth */ true, /* num_cpu_devices */ 2), TF_DeleteBuffer); TFE_ContextOptionsSetConfig(opts.get(), config->data, config->length, status.get()); std::unique_ptr<TFE_Context, decltype(&TFE_DeleteContext)> context( TFE_NewContext(opts.get(), status.get()), TFE_DeleteContext);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jul 08 23:47:35 GMT 2021 - 15.3K bytes - Viewed (0) -
tensorflow/c/experimental/filesystem/plugins/posix/posix_filesystem.cc
plugin_memory_allocate(TF_READ_ONLY_MEMORY_REGION_OPS_SIZE)); ops->read_only_memory_region_ops->cleanup = tf_read_only_memory_region::Cleanup; ops->read_only_memory_region_ops->data = tf_read_only_memory_region::Data; ops->read_only_memory_region_ops->length = tf_read_only_memory_region::Length; ops->filesystem_ops = static_cast<TF_FilesystemOps*>( plugin_memory_allocate(TF_FILESYSTEM_OPS_SIZE));
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Sun Mar 24 20:08:23 GMT 2024 - 15.8K bytes - Viewed (0) -
tensorflow/c/eager/dlpack.cc
owner->reference.Unref(); delete owner; } // Converts TF_DATAType to DLPack data type. DLDataType GetDlDataType(TF_DataType data_type, TF_Status* status) { DLDataType dtype; dtype.lanes = 1; dtype.bits = TF_DataTypeSize(data_type) * 8; switch (data_type) { case TF_DataType::TF_BOOL: dtype.code = DLDataTypeCode::kDLBool; break; case TF_DataType::TF_HALF:
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 12.8K bytes - Viewed (0)