- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 12 for elseif (0.17 sec)
-
tensorflow/c/eager/dlpack.cc
} ctx.device_id = device_id; if (device_type == "CPU") { ctx.device_type = DLDeviceType::kDLCPU; } else if (device_type == "GPU") { #if TENSORFLOW_USE_ROCM ctx.device_type = DLDeviceType::kDLROCM; #else ctx.device_type = DLDeviceType::kDLCUDA; #endif } else { status->status = tensorflow::errors::InvalidArgument( "Unsupported Device Type for dlpack"); }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 12.8K bytes - Viewed (0) -
tensorflow/c/eager/gradients.cc
TensorShape::MaxDimensions(), "."); } TensorShapeProto proto; if (num_dims < 0) { proto.set_unknown_rank(true); } else { for (int d = 0; d < num_dims; ++d) { proto.add_dim()->set_size(dims[d]); } } forward_op_->attrs.Set(attr_name, proto); return op_->SetAttrShape(attr_name, dims, num_dims); }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 19.3K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/math_grad.cc
auto dtype = input->DataType(); if (DataTypeIsFloating(BaseType(dtype)) || DataTypeIsInteger(BaseType(dtype))) { return tensorflow::ops::Identity(ctx, input, output, name); } else if (!DataTypeIsComplex(BaseType(dtype)) && BaseType(dtype) != DT_VARIANT) { return errors::InvalidArgument( "Expected numeric or variant tensor, got dtype ", dtype); }
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 15.2K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental_graph.cc
const FullTypeDef* ft; mutex_lock l(graph_->mu); graph_->graph.NodeType(output_.oper->node.name(), &ft); if (ft == nullptr) { return FullTypeDef(); } else { return *ft; } } TF_Output output_; // For LLVM style RTTI. static bool classof(const AbstractTensorHandle* ptr) { return ptr->getKind() == kGraph; } private:
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/c_api.cc
bool clear_existing_contexts) { #if defined(IS_MOBILE_PLATFORM) status->status = tensorflow::errors::Unimplemented( "TFE_ContextSetServerDef not supported on mobile"); #else // !defined(IS_MOBILE_PLATFORM) tensorflow::ServerDef server_def; if (!server_def.ParseFromArray(proto, proto_len)) { status->status = tensorflow::errors::InvalidArgument(
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Mar 12 20:00:09 GMT 2024 - 43.9K bytes - Viewed (2) -
tensorflow/c/eager/c_api_experimental.cc
if (op_to_reset) { tensorflow::ImmediateExecutionOperation* op = tensorflow::unwrap(op_to_reset); op->Clear(); status->status = op->Reset(op_or_function_name, raw_device_name); } else { TF_SetStatus(status, TF_INVALID_ARGUMENT, "op_to_reset should not be nullptr"); } } void TFE_ContextEnableGraphCollection(TFE_Context* ctx) {
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/filesystem/plugins/posix/posix_filesystem.cc
offset += static_cast<uint64_t>(r); n -= r; // safe as 0 < r <= n so n will never underflow read += r; } else if (r == 0) { TF_SetStatus(status, TF_OUT_OF_RANGE, "Read fewer bytes than requested"); break; } else if (errno == EINTR || errno == EAGAIN) { // Retry } else { TF_SetStatusFromIOError(status, errno, posix_file->filename); break; } } return read; }
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Sun Mar 24 20:08:23 GMT 2024 - 15.8K bytes - Viewed (0) -
tensorflow/c/eager/tape.h
zero_indices.push_back(i); } } } else { any_gradient_nonzero = true; Gradient* new_gradients = nullptr; if (grad_it->second.size() == 1) { new_gradients = grad_it->second.at(0); } else { new_gradients = vspace.AggregateGradients(grad_it->second); }
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/c_api_experimental.cc
if (it != loaded_libs->end()) { lib_handle->lib_handle = it->second; } else { status->status = env->LoadDynamicLibrary(library_filename, &lib_handle->lib_handle); if (status->status.ok()) { TF_CHECK_OK( tensorflow::RegisterPluggableDevicePlugin(lib_handle->lib_handle)); } else { delete lib_handle; return nullptr; } }
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/eager/parallel_device/parallel_device_lib.cc
execution_state_ == ExecutionState::kHasResult) { start_execute_.wait(l); } if (execution_state_ == ExecutionState::kShuttingDown) { return; } else if (execution_state_ == ExecutionState::kReadyToExecute) { // op_outputs_ may have been std::moved op_outputs_ = std::vector<TensorHandlePtr>();
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)