- Sort Score
- Result 10 results
- Languages All
Results 21 - 29 of 29 for if (0.19 sec)
-
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/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/checkpoint_reader.cc
tensorflow::TensorShape shape; status = v2_reader_->LookupDtypeAndShape(name, &dtype, &shape); if (status.ok()) { out_tensor->reset(new Tensor(dtype, shape)); status = v2_reader_->Lookup(name, out_tensor->get()); if (!status.ok()) out_tensor->reset(); } } if (!status.ok()) { tsl::Set_TF_Status_from_Status(out_status, status); } }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Aug 25 21:29:12 GMT 2023 - 5.5K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/tape/tape_operation.cc
return parent_op_->SetDeviceName(name); } Status TapeOperation::AddInput(AbstractTensorHandle* input) { TF_RETURN_IF_ERROR(parent_op_->AddInput(input)); forward_op_.inputs.push_back(input); return OkStatus(); } Status TapeOperation::AddInputList( absl::Span<AbstractTensorHandle* const> inputs) { TF_RETURN_IF_ERROR(parent_op_->AddInputList(inputs)); for (auto input : inputs) { forward_op_.inputs.push_back(input); }
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Tue Jun 07 01:53:35 GMT 2022 - 9K bytes - Viewed (1) -
tensorflow/c/eager/c_api_debug.cc
std::vector<int64_t> shape; int rank = -1; *status = handle.NumDims(&rank); if (!status->ok()) { return shape; } shape.reserve(rank); for (int i = 0; i < rank; ++i) { int64_t dim; *status = handle.Dim(i, &dim); if (!status->ok()) { return shape; } shape.push_back(dim); } return shape; } } // namespace
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Aug 11 01:20:50 GMT 2021 - 2.5K bytes - Viewed (0) -
tensorflow/c/experimental/filesystem/plugins/posix/copy_file_portable.cc
size_t chunk = size - offset; if (chunk > kPosixCopyFileBufferSize) chunk = kPosixCopyFileBufferSize; rc = read(src_fd, buffer.get(), chunk); if (rc < 0) return -1; int total_write = 0; int total_read = rc; while (total_write < total_read && rc > 0) { rc = write(dst_fd, buffer.get() + total_write, chunk - total_write); if (rc < 0) return -1; total_write += rc;
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Fri Nov 22 21:23:55 GMT 2019 - 1.9K 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) -
src/cmd/cgo/internal/swig/testdata/callback/main.cc
C++ - Registered: Tue Apr 23 11:13:09 GMT 2024 - Last Modified: Fri May 12 12:00:07 GMT 2023 - 384 bytes - Viewed (0)