- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 11 for v_size (0.17 sec)
-
tensorflow/c/eager/c_api_unified_experimental.cc
TF_Status* s) { unwrap(o)->expected_num_outputs = num_outputs; unwrap(o)->outputs.clear(); unwrap(o)->outputs.resize(num_outputs); } int TF_OutputListNumOutputs(TF_OutputList* o) { return unwrap(o)->outputs.size(); } TF_AbstractTensor* TF_OutputListGet(TF_OutputList* o, int i) { return wrap(unwrap(o)->outputs[i]); }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 9K bytes - Viewed (0) -
tensorflow/c/c_api.cc
DCHECK(tf_results->missing_unused_key_names_data.empty()); size_t size = results.missing_unused_input_map_keys.size(); tf_results->missing_unused_key_names.resize(size); tf_results->missing_unused_key_indexes.resize(size); for (int i = 0; i < size; ++i) { TensorId id = results.missing_unused_input_map_keys[i]; tf_results->missing_unused_key_names_data.emplace_back(id.first); tf_results->missing_unused_key_names[i] =
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) -
tensorflow/c/experimental/filesystem/plugins/posix/posix_filesystem.cc
#include "tensorflow/c/tf_status.h" // Implementation of a filesystem for POSIX environments. // This filesystem will support `file://` and empty (local) URI schemes. static void* plugin_memory_allocate(size_t size) { return calloc(1, size); } static void plugin_memory_free(void* ptr) { free(ptr); } // SECTION 1. Implementation for `TF_RandomAccessFile` // ----------------------------------------------------------------------------
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/c_api_experimental.cc
static_cast<bool>(should_enable); } void TF_SetXlaMinClusterSize(int size) { tensorflow::MarkForCompilationPassFlags* flags = tensorflow::GetMarkForCompilationPassFlags(); flags->tf_xla_min_cluster_size = size; } TF_Buffer* TF_CreateConfig(unsigned char enable_xla_compilation, unsigned char gpu_memory_allow_growth,
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/experimental/gradients/grad_test_helper.cc
AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> inputs, absl::Span<AbstractTensorHandle*> outputs) -> Status { Tape tape(/*persistent=*/false); for (size_t i{}; i < inputs.size(); ++i) { tape.Watch(inputs[i]); } std::vector<AbstractTensorHandle*> temp_outputs(1); AbstractContextPtr tape_ctx(new TapeContext(ctx, &tape, grad_registry)); TF_RETURN_IF_ERROR(
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/custom_gradient_test.cc
Status Compute(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> grad_outputs, absl::Span<AbstractTensorHandle*> grad_inputs) override { CHECK_EQ(grad_outputs.size(), 1); CHECK_EQ(grad_inputs.size(), 1); grad_inputs[0] = grad_outputs[0]; if (grad_inputs[0]) { grad_inputs[0]->Ref(); } return absl::OkStatus(); } }; // Computes: // // @tf.custom_gradient
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 4.8K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/array_grad.cc
public: Status Compute(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> grad_outputs, absl::Span<AbstractTensorHandle*> grad_inputs) override { for (int i = 0; i < grad_outputs.size(); i++) { auto grad_input = grad_outputs[i]; // TODO(srbs): Should we add a copy contructor to AbstractTensorHandle // that takes care of this similar to `Tensor`? if (grad_input) {
C++ - Registered: Tue Apr 09 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 1.6K bytes - Viewed (0) -
tensorflow/c/eager/dlpack.cc
std::vector<int64_t>* shape_arr = &tf_dlm_tensor_ctx->shape; std::vector<int64_t>* stride_arr = &tf_dlm_tensor_ctx->strides; shape_arr->resize(ndim); stride_arr->resize(ndim, 1); for (int i = 0; i < ndim; i++) { (*shape_arr)[i] = tensor->dim_size(i); } for (int i = ndim - 2; i >= 0; --i) { (*stride_arr)[i] = (*shape_arr)[i + 1] * (*stride_arr)[i + 1]; }
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/parallel_device/parallel_device_lib.cc
for (int device_index = 0; device_index < underlying_devices_.size(); ++device_index) { DeviceThread* device_thread = device_threads_[device_index].get(); std::vector<TFE_TensorHandle*> device_inputs; device_inputs.reserve(inputs.size()); for (int input_index = 0; input_index < inputs.size(); ++input_index) { // Parallel tensors are divided between operations by device.
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/unified_api_testutil.cc
} int retvals = outputs.size() - null_indices.size(); std::vector<AbstractTensorHandle*> fn_outputs(retvals); TF_RETURN_IF_ERROR(fn_op->Execute( absl::Span<AbstractTensorHandle*>(fn_outputs.data(), fn_outputs.size()), &retvals)); int skipped_indices = 0; for (int i = 0; i < outputs.size(); i++) { if (!null_indices.contains(i)) {
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)