- Sort Score
- Result 10 results
- Languages All
Results 11 - 15 of 15 for what (0.18 sec)
-
tensorflow/c/eager/unified_api_test.cc
} public: bool UseMlir() const { return strcmp(std::get<0>(GetParam()), "mlir") == 0; } bool UseFunction() const { return std::get<2>(GetParam()); } }; // Checks that inputs[0] is a scalar. Status TestScalarShape(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> inputs, absl::Span<AbstractTensorHandle*> outputs) {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Feb 27 13:57:45 GMT 2024 - 6.7K bytes - Viewed (0) -
tensorflow/c/eager/c_api_experimental.cc
(*sess_options.config.mutable_device_count())["CPU"] = num_cpus; status->status = tensorflow::DeviceFactory::AddCpuDevices(sess_options, prefix, &devices); // Remove the device that has the host device name since host device is alreay // in an initialized context. for (auto d = devices.begin(); d != devices.end();) { if (absl::StrContains(d->get()->name(), "CPU:0")) { d = devices.erase(d);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Apr 11 23:52:39 GMT 2024 - 35.9K bytes - Viewed (3) -
tensorflow/c/eager/c_api.cc
buf->Unref(); return tensorflow::wrap(tensorflow::TensorHandle::CreateLocalHandle( std::move(t), device, device, context)); } // This function will block till the operation that produces `h` has // completed. This is only valid on local TFE_TensorHandles. Returns the size in // bytes of the memory pointed to by the device pointer returned above. size_t TFE_TensorHandleDeviceMemorySize(TFE_TensorHandle* h,
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Mar 12 20:00:09 GMT 2024 - 43.9K bytes - Viewed (2) -
tensorflow/c/c_api_test.cc
// Fetch and see that the new value is returned. TF_GraphGetTensorShape(graph, feed_out_0, returned_dims, num_dims, s); ASSERT_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s); EXPECT_EQ(dims[0], returned_dims[0]); EXPECT_EQ(dims[1], returned_dims[1]); // Try to set 'unknown' with unknown rank on the shape and see that // it doesn't change.
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 96.9K bytes - Viewed (3) -
tensorflow/c/experimental/gradients/nn_grad_test.cc
TF_RETURN_IF_ERROR(ops::SparseSoftmaxCrossEntropyWithLogits( ctx, inputs[0], inputs[1], &loss, &backprop, "SparseSoftmaxCrossEntropyWithLogits")); // `gradient_checker` only works with model that returns only 1 tensor. // Although, `ops::SparseSoftmaxCrossEntropyWithLogits` returns 2 tensors, the // second tensor isn't needed for computing gradient so we could safely drop // it. outputs[0] = loss;
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 8.3K bytes - Viewed (0)