- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for zeros (0.14 sec)
-
tensorflow/c/eager/tape.h
return absl::OkStatus(); } // We may need to allocate zero inputs for trainable dtypes we don't have JVPs // for. Make sure they get cleaned up. std::vector<Gradient*> new_zeros; auto delete_new_zeros = gtl::MakeCleanup([&new_zeros, this] { for (Gradient* tensor : new_zeros) { this->vspace_.DeleteGradient(tensor); } }); std::vector<Gradient*> in_grads;
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/eager/gradients.cc
} } TF_RETURN_IF_ERROR(GradientTape::ComputeGradient( vspace, target_tensor_ids, source_tensor_ids, sources_that_are_targets, output_gradients, result, /*build_default_zeros_grads*/ false)); return absl::OkStatus(); } // Helper functions which delegate to `AbstractOperation`, update // the state of the ForwardOperation and call the tape as appropriate.
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/eager/dlpack.cc
void* data = dl_tensor->data; if (dl_tensor->byte_offset != 0) { status->status = tensorflow::errors::InvalidArgument( "Unsupported byte_offset (", dl_tensor->byte_offset, ") from DLPack, must be zero"); return nullptr; } size_t total_bytes = dl_tensor->dtype.bits / 8; for (int i = 0; i < num_dims; i++) { total_bytes *= dims[i]; } if (dl_tensor->strides != nullptr &&
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/c_api_experimental.h
unsigned char should_enable); // Create a serialized tensorflow.ConfigProto proto, where: // // a) ConfigProto.optimizer_options.global_jit_level is set to ON_1 if // `enable_xla_compilation` is non-zero, and OFF otherwise. // b) ConfigProto.gpu_options.allow_growth is set to `gpu_memory_allow_growth`. // c) ConfigProto.device_count is set to `num_cpu_devices`. TF_CAPI_EXPORT extern TF_Buffer* TF_CreateConfig(
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Apr 27 21:07:00 GMT 2023 - 15.1K bytes - Viewed (0) -
tensorflow/c/eager/c_api_cluster_test.cc
TF_DeleteTensor(t); for (int i = 0; i < expected_values.size(); i++) { EXPECT_EQ(expected_values[i], actual_values[i]) << "Mismatch in expected values at (zero-based) index " << i; } } void CheckRemoteMatMulExecutesOK(TFE_Context* ctx, const char* remote_device_name, const char* local_device_name) {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Apr 14 10:03:59 GMT 2023 - 19.3K bytes - Viewed (0) -
ci/official/containers/linux_arm64/builder.devtoolset/stringop_trunc.patch
Others - Registered: Tue May 07 12:40:20 GMT 2024 - Last Modified: Mon Sep 18 14:52:45 GMT 2023 - 42.9K bytes - Viewed (1) -
tensorflow/c/eager/parallel_device/parallel_device.cc
return result; } // Used as an argument to TFE_NewCustomDeviceTensorHandle, indicating how // ParallelTensors wrapped in TFE_TensorHandles should be cleaned up once their // reference counts drop to zero. void ParallelTensorDeallocator(void* data) { delete reinterpret_cast<ParallelTensor*>(data); } // Used as an argument to TFE_NewCustomDeviceTensorHandle, for computing the
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Mar 29 22:05:31 GMT 2023 - 18.3K bytes - Viewed (0)