- Sort Score
- Result 10 results
- Languages All
Results 11 - 17 of 17 for TF_RETURN_IF_ERROR (0.27 sec)
-
tensorflow/c/eager/parallel_device/parallel_device_lib.cc
if (!shape_.has_value()) { TF_Status status; PartialTensorShape combined_shape; TF_RETURN_IF_ERROR(unwrap(tensors_[0].get())->Shape(&combined_shape)); for (const TensorHandlePtr& component : tensors_) { PartialTensorShape component_shape; TF_RETURN_IF_ERROR(unwrap(component.get())->Shape(&component_shape)); if (combined_shape.dims() < 0 ||
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Feb 09 07:47:20 GMT 2024 - 25.4K bytes - Viewed (1) -
tensorflow/c/eager/gradients_test.cc
Status s = StatusFromTF_Status(status.get()); CHECK_EQ(errors::OK, s.code()) << s.message(); } }; Status RegisterGradients(GradientRegistry* registry) { TF_RETURN_IF_ERROR(RegisterNotDifferentiable(registry, "CheckNumerics")); return absl::OkStatus(); } TEST_P(CppGradients, TestSetAttrString) { std::unique_ptr<TF_Status, decltype(&TF_DeleteStatus)> status(
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/eager/unified_api_test.cc
Status TestScalarShape(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> inputs, absl::Span<AbstractTensorHandle*> outputs) { PartialTensorShape shape; TF_RETURN_IF_ERROR(inputs[0]->Shape(&shape)); if (shape.dims() != 0) { return errors::InvalidArgument( "Tensor expected to have scalar shape found rank: ", shape.dims()); } return absl::OkStatus(); }
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/tape.h
// backward function under a gradient tape. forward_grads.resize(output_tensors.size()); TF_RETURN_IF_ERROR(ForwardpropFromTape( op_type, output_tensors, backward_function_getter, backward_function_deleter, in_grads, absl::MakeSpan(forward_grads))); } else { TF_RETURN_IF_ERROR( (*forward_function)(in_grads, &forward_grads, use_batch_)); }
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Apr 02 12:40:29 GMT 2024 - 47.2K bytes - Viewed (1) -
tensorflow/c/experimental/gradients/array_grad_test.cc
absl::Span<AbstractTensorHandle* const> inputs, absl::Span<AbstractTensorHandle*> outputs) { std::vector<AbstractTensorHandle*> temp_outputs(2); TF_RETURN_IF_ERROR( ops::IdentityN(ctx, inputs, absl::MakeSpan(temp_outputs), "IdentityN")); // Although, `ops::IdentityN` returns 2 tensors, the first 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 - 5K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/nn_grad_test.cc
AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> inputs, absl::Span<AbstractTensorHandle*> outputs) { AbstractTensorHandle* loss; 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.
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.cc
for (int i = 0; i < nreturn_nodes; ++i) { opts.return_tensors.push_back(ToTensorId(nodes_to_return[i])); } // TODO(skyewm): change to OutputTensor tensorflow::ImportGraphDefResults results; TF_RETURN_IF_ERROR( ImportGraphDef(opts, gdef, dst_graph, dst_refiner, &results)); for (const auto& pair : results.return_tensors) { return_nodes->emplace_back(pair.first, pair.second); }
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)