- Sort Score
- Result 10 results
- Languages All
Results 1 - 8 of 8 for checkID (0.24 sec)
-
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/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/experimental/gradients/grad_test_helper.cc
limitations under the License. ==============================================================================*/ #include "tensorflow/c/experimental/gradients/grad_test_helper.h" #include "tensorflow/c/eager/gradient_checker.h" #include "tensorflow/c/experimental/gradients/tape/tape_context.h" #include "tensorflow/core/platform/test.h" namespace tensorflow { namespace gradients { namespace internal {
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 5K 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/c_api_function.cc
"' has a reference type ", DataTypeString(dt)) : absl::OkStatus(); } // Converts `ninputs` and `inputs` into `inputs_tensors` and `input_nodes` and // does various checks while doing so. `input_nodes` will contain the same // information as input_tensors just in a different structure to make // following processing easier. TODO(iga): Simplify this nested structure. Status ProcessInputs(
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 13.6K bytes - Viewed (2) -
tensorflow/c/eager/unified_api_test.cc
Status s = StatusFromTF_Status(status.get()); CHECK_EQ(errors::OK, s.code()) << s.message(); } 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,
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/experimental/gradients/nn_grad_test.cc
AbstractTensorHandle* backprop; 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.
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) -
tensorflow/c/c_api_test.cc
TF_DeviceList* device_list = list.get(); CHECK_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s); const int num_devices = TF_DeviceListCount(device_list); LOG(INFO) << "There are " << num_devices << " devices."; for (int i = 0; i < num_devices; ++i) { const char* device_name = TF_DeviceListName(device_list, i, s); CHECK_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s);
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)