- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 23 for Check (0.15 sec)
-
tensorflow/c/experimental/gradients/custom_gradient_test.cc
CHECK_EQ(errors::OK, s.code()) << s.message(); } }; class PassThroughGradientFunction : public GradientFunction { public: Status Compute(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> grad_outputs, absl::Span<AbstractTensorHandle*> grad_inputs) override { CHECK_EQ(grad_outputs.size(), 1); CHECK_EQ(grad_inputs.size(), 1);
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 4.8K bytes - Viewed (0) -
tensorflow/c/c_api.cc
handle_data.set_is_set(true); { mutex_lock l(graph->mu); tensorflow::shape_inference::InferenceContext* ic = graph->refiner.GetContext(node); CHECK(ic != nullptr); // Crash OK CHECK_LT(output.index, ic->num_outputs()); // Crash OK const auto* shapes_and_types = ic->output_handle_shapes_and_types(output.index); if (shapes_and_types == nullptr) return nullptr;
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/c_api_experimental.cc
// a different TFE context for each thread of execution (for running graph // functions, and their send/recvs corountines). config.set_inter_op_parallelism_threads(1); TF_Buffer* ret = TF_NewBuffer(); TF_CHECK_OK(MessageToBuffer(config, ret)); return ret; } TF_Buffer* TF_CreateRunOptions(unsigned char enable_full_trace) { tensorflow::RunOptions options; if (enable_full_trace) {
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/c_api_cluster_test.cc
EXPECT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); // Create a new tensor_handle. TFE_TensorHandle* h0_task0_new = TestMatrixTensorHandle(ctx); // Check that copying it to the old remote device (named localhost) fails. TFE_TensorHandleCopyToDevice(h0_task0_new, ctx, remote_device_name, status); EXPECT_NE(TF_OK, TF_GetCode(status)) << TF_Message(status);
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/eager/c_api_experimental_reader_test.cc
auto* counter = CreateCounter0(counter_name); auto* reader = TFE_MonitoringNewCounterReader(counter_name); IncrementCounter0(counter); int64_t actual = TFE_MonitoringReadCounter0(reader); CHECK_EQ(actual, 1); } TEST(CAPI, MonitoringCellReader1) { auto counter_name = "test/counter1"; auto label_name = "test/label"; auto* counter = CreateCounter1(counter_name, label_name);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Apr 20 03:14:47 GMT 2023 - 2.9K bytes - Viewed (0) -
tensorflow/c/eager/gradients_test.cc
t.reset(x_raw); } AbstractOperationPtr check_numerics_op(ctx->CreateOperation()); ForwardOperation forward_op; Status s = Reset(check_numerics_op.get(), "CheckNumerics", /*raw_device_name=*/nullptr, &forward_op); ASSERT_EQ(errors::OK, s.code()) << s.message(); if (isa<TracingOperation>(check_numerics_op.get())) { s = dyn_cast<TracingOperation>(check_numerics_op.get()) ->SetOpName("check_numerics");
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 7K bytes - Viewed (0) -
tensorflow/c/experimental/next_pluggable_device/tensor_pjrt_buffer_util_test.cc
using ::testing::NotNull; using ::tsl::testing::StatusIs; PJRT_Buffer* CreateCBuffer() { auto status = pjrt::PjrtApi(DEVICE_CPU); if (!status.ok()) { CHECK_OK(pjrt::SetPjrtApi(DEVICE_CPU, GetPjrtApi())); } auto pjrt_client = xla::GetCApiClient(DEVICE_CPU); CHECK_OK(pjrt_client.status()); auto c_api_client = down_cast<xla::PjRtCApiClient*>(pjrt_client->get()); std::vector<int32_t> data(1, 0);
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Mon Oct 30 19:20:20 GMT 2023 - 7.2K bytes - Viewed (0) -
tensorflow/c/experimental/grappler/grappler_test.cc
TF_ASSERT_OK(cluster->Provision()); TrivialTestGraphInputYielder fake_input(4, 1, 10, false, cluster->GetDeviceNames()); GrapplerItem item; CHECK(fake_input.NextItem(&item)); TF_Status* status = TF_NewStatus(); TF_GraphProperties* graph_properties = TF_NewGraphProperties(reinterpret_cast<TF_GrapplerItem*>(&item));
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Thu Apr 13 22:30:58 GMT 2023 - 11.6K bytes - Viewed (0) -
tensorflow/c/eager/c_api_distributed_test.cc
namespace { using ::tensorflow::string; // Add the values of three variables on three different tasks. string AddVariablesFunction() { tensorflow::FunctionDef def; CHECK(tensorflow::protobuf::TextFormat::ParseFromString( " signature {" " name: 'AddVariablesFunction'" " input_arg {" " name: 'var'" " type: DT_RESOURCE" " }"
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 23.5K bytes - Viewed (0) -
tensorflow/c/eager/gradient_checker.cc
// Initialize auxilary data structures. vector<float> thetaPlus_data(num_elems); vector<float> thetaMinus_data(num_elems); AbstractTensorHandle* f_outputs[1]; // Numerical Grad Check for (int i = 0; i < num_elems; i++) { // Get relative epsilon value float epsilon = theta_data[i] == 0 ? 1e-4 : std::abs(theta_data[i] * 1e-4); AbstractTensorHandlePtr two_eps; {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 7.3K bytes - Viewed (0)