- Sort Score
- Result 10 results
- Languages All
Results 51 - 60 of 60 for if (0.11 sec)
-
tensorflow/c/eager/c_api_unified_experimental_graph.cc
TracingOperationPtr operation(CreateOperation()); TF_RETURN_IF_ERROR(operation->Reset("Placeholder", nullptr)); TF_RETURN_IF_ERROR( operation->SetOpName(absl::StrCat("_input_", inputs_.size()).c_str())); TF_RETURN_IF_ERROR(operation->SetAttrType("dtype", dtype)); if (!shape.unknown_rank()) { TF_RETURN_IF_ERROR(operation->SetAttrShape(
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Mar 12 20:00:09 GMT 2024 - 15.4K bytes - Viewed (1) -
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/eager/c_api_experimental.cc
dimvec[i] = static_cast<int64_t>(dims[i]); } if (ctx == nullptr) { status->status = tensorflow::errors::InvalidArgument("Invalid Context"); return nullptr; } tensorflow::AbstractTensorInterface* t = tensorflow::unwrap(ctx)->CreateTensor( static_cast<tensorflow::DataType>(dtype), dimvec); if (t == nullptr) { status->status =
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/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) -
tensorflow/c/eager/tracing_utils.cc
namespace tensorflow { namespace tracing { Status MaybeSetOpName(AbstractOperation* op, const char* op_name) { if (isa<TracingOperation>(op)) { TF_RETURN_IF_ERROR(dyn_cast<TracingOperation>(op)->SetOpName(op_name)); } if (isa<gradients::TapeOperation>(op)) { TF_RETURN_IF_ERROR(MaybeSetOpName( dyn_cast<gradients::TapeOperation>(op)->GetBackingOperation(), op_name)); }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Feb 27 13:57:45 GMT 2024 - 1.4K bytes - Viewed (0) -
tensorflow/c/eager/c_api_remote_test_util.cc
} else { // Handles are on task0 (local), and task2, but op is on task1. matmul = MatMulOp(ctx, h0_task0, h1_task2); } if (remote) { TFE_OpSetDevice(matmul, task1_name, status); ASSERT_EQ(TF_GetCode(status), TF_OK) << TF_Message(status); } else if (!async) { // Set the local device to CPU to easily validate mirroring string cpu_device_name;
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Dec 11 22:56:03 GMT 2020 - 9.1K bytes - Viewed (0) -
tensorflow/c/eager/c_api_test_util.cc
if (TF_GetCode(status) != TF_OK) return nullptr; TFE_OpSetAttrType(op, "dtype", TF_FLOAT); TFE_OpSetAttrShape(op, "shape", {}, 0, status); TFE_OpSetAttrString(op, "container", "localhost", 0); TFE_OpSetAttrString(op, "shared_name", "", 0); if (!device_name.empty()) { TFE_OpSetDevice(op, device_name.c_str(), status); } if (TF_GetCode(status) != TF_OK) return nullptr;
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Feb 21 22:37:46 GMT 2024 - 23.5K bytes - Viewed (2) -
tensorflow/c/experimental/filesystem/plugins/posix/copy_file_linux.cc
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Fri Nov 22 21:23:55 GMT 2019 - 1.5K bytes - Viewed (0) -
tensorflow/c/c_api_experimental_test.cc
TF_DeleteTensor(tensor_1X1X6); TF_DeleteTensor(tensor_1X6); } TEST(CAPI_EXPERIMENTAL, LibraryPluggableDeviceLoadFunctions) { // TODO(penpornk): Enable this test on Windows. #if !defined(PLATFORM_WINDOWS) #if !defined(TENSORFLOW_NO_SHARED_OBJECTS) // Load the library. TF_Status* status = TF_NewStatus(); string lib_path = tensorflow::GetDataDependencyFilepath(tensorflow::io::JoinPath(
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Jan 17 22:27:52 GMT 2023 - 13.1K bytes - Viewed (1) -
tensorflow/c/c_api_function_test.cc
ASSERT_TRUE(a_edges.find(e) != a_edges.end()) << "Failed to find expected control edge " << e.ToString() << " in fdef: " << fdef.DebugString(); } // If caller specified all edges, check that we have seen all if (is_exact_edges) { ASSERT_EQ(e_edges.size() + c_edges.size(), a_edges.size()) << "Expected edges: " << ToString(e_edges) << " Expected Control edges: " << ToString(c_edges)
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jul 20 22:08:54 GMT 2023 - 63.6K bytes - Viewed (6)