- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 29 for if (0.48 sec)
-
tensorflow/c/experimental/filesystem/plugins/posix/posix_filesystem_static.cc
// Return value will be unused bool StaticallyRegisterLocalFilesystems() { TF_FilesystemPluginInfo info; TF_InitPlugin(&info); Status status = filesystem_registration::RegisterFilesystemPluginImpl(&info); if (!status.ok()) { VLOG(0) << "Static POSIX filesystem could not be registered: " << status; return false; } return true; } // Perform the actual registration
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Sun Mar 24 20:08:23 GMT 2024 - 1.5K bytes - Viewed (0) -
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/eager/gradient_checker.cc
// Run the model. TF_RETURN_IF_ERROR( RunModel(forward, ctx, inputs, model_outputs, use_function)); AbstractTensorHandlePtr model_out(model_outputs[0]); TF_Tensor* model_out_tensor; TF_RETURN_IF_ERROR(GetValue(model_out.get(), &model_out_tensor)); int num_dims_out = TF_NumDims(model_out_tensor); TF_DeleteTensor(model_out_tensor); // If the output is a scalar, then return the scalar output
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) -
tensorflow/c/eager/custom_device_testutil.cc
if (TF_GetCode(s) != TF_OK) return; int num_inputs = TFE_OpGetFlatInputCount(original_op, s); if (TF_GetCode(s) != TF_OK) return; for (int j = 0; j < num_inputs; ++j) { TFE_TensorHandle* input = TFE_OpGetFlatInput(original_op, j, s); if (TF_GetCode(s) != TF_OK) return; const char* input_device = TFE_TensorHandleDeviceName(input, s); if (TF_GetCode(s) != TF_OK) return;
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Mar 03 20:47:31 GMT 2021 - 8.3K bytes - Viewed (0) -
tensorflow/c/experimental/next_pluggable_device/tensor_pjrt_buffer_util_test.cc
namespace tensorflow { namespace { using ::testing::HasSubstr; 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());
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/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/abstract_tensor_handle.cc
#include "tensorflow/c/eager/abstract_tensor_handle.h" namespace tensorflow { std::string AbstractTensorHandle::DebugString() const { PartialTensorShape shape; Status s = Shape(&shape); std::string shape_string; if (!s.ok()) { shape_string = "<error computing shape>"; } else { shape_string = shape.DebugString(); } return absl::StrCat("TensorHandle(shape=", shape_string,
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 1.4K bytes - Viewed (0) -
tensorflow/c/eager/unified_api_testutil.cc
for (auto output : output_list.outputs) { output->Unref(); } TF_RETURN_IF_ERROR(ctx->RegisterFunction(func)); } AbstractOperationPtr fn_op(ctx->CreateOperation()); TF_RETURN_IF_ERROR(fn_op->Reset(fn_name, /*raw_device_name=*/nullptr)); for (auto input : inputs) { TF_RETURN_IF_ERROR(fn_op->AddInput(input)); } int retvals = outputs.size() - null_indices.size();
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Feb 27 13:57:45 GMT 2024 - 5.7K bytes - Viewed (0) -
tensorflow/c/eager/unified_api_test.cc
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(); } TEST_P(UnifiedAPI, TestTensorShapeScalar) { if (UseFunction() && UseMlir()) { // TODO(b/173074167): Remove this.
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/next_pluggable_device/tensor_pjrt_buffer_util.cc
tensorflow::AsyncValueTensor* av_tensor = tensorflow::AsyncValueTensor::FromTensor(tensor); if (av_tensor == nullptr || av_tensor->GetBuffer() == nullptr) { return absl::InternalError("Input tensor does not have PjRtBuffer."); } auto* c_api_buffer = dynamic_cast<xla::PjRtCApiBuffer*>(av_tensor->GetBuffer().get()); if (c_api_buffer == nullptr) { return absl::InternalError(
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Mon Oct 30 19:20:20 GMT 2023 - 3.7K bytes - Viewed (0)