- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 30 for arg (0.12 sec)
-
tensorflow/c/c_test_util.cc
static void BoolDeallocator(void* data, size_t, void* arg) { delete[] static_cast<bool*>(data); } static void Int32Deallocator(void* data, size_t, void* arg) { delete[] static_cast<int32_t*>(data); } static void DoubleDeallocator(void* data, size_t, void* arg) { delete[] static_cast<double*>(data); } static void FloatDeallocator(void* data, size_t, void* arg) { delete[] static_cast<float*>(data); }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Oct 15 03:16:52 GMT 2021 - 17.8K bytes - Viewed (2) -
ci/official/containers/linux_arm64/Dockerfile
RUN dpkg -i /patchelf/patchelf_0.14.3-1_arm64.deb # All lines past this point are reset when $CACHEBUSTER is set. We need this # for Python specifically because we install some nightly packages which are # likely to change daily. ARG CACHEBUSTER=0 RUN echo $CACHEBUSTER # Setup build and environment COPY devel.usertools /usertools COPY devel.bashrc /root/.bashrc COPY ld.so.conf /dt10/etc/ # Setup JAX Python environment. FROM devel as jax
Plain Text - Registered: Tue May 07 12:40:20 GMT 2024 - Last Modified: Mon Jan 08 09:32:19 GMT 2024 - 4.1K bytes - Viewed (1) -
tensorflow/c/eager/dlpack.cc
if (!status->status.ok()) { return nullptr; } return tensor; } // Deleter for DLManagedTensor void DLManagedTensorDeleter(DLManagedTensor* arg) { TfDlManagedTensorCtx* owner = static_cast<TfDlManagedTensorCtx*>(arg->manager_ctx); owner->reference.Unref(); delete owner; } // Converts TF_DATAType to DLPack data type. DLDataType GetDlDataType(TF_DataType data_type, TF_Status* status) {
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/c_api_unified_experimental_test.cc
// Build two abstract input tensors as function arguments. std::vector<TF_AbstractTensor*> func_args; { TFE_Context* eager_ctx = TF_ExecutionContextGetTFEContext(eager_execution_ctx, s); // 1st Arg float vals1[] = {0.0f, 1.0f, 1.0f, 0.0f}; int64_t dims[] = {2, 2}; // Matrices will be 2 x 2 int num_dims = sizeof(dims) / sizeof(dims[0]); TFE_TensorHandle* input_eager =
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri May 19 21:44:52 GMT 2023 - 39.1K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.cc
if (!status->status.ok()) return nullptr; if (input_index >= op_def->input_arg_size() || input_index < 0) { status->status = tensorflow::errors::InvalidArgument( input_index, " out of range for ", op_name); return nullptr; } const tensorflow::OpDef_ArgDef& input_arg = op_def->input_arg()[input_index]; if (input_arg.number_attr().empty()) { status->status = tensorflow::errors::NotFound(
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) -
ci/official/containers/linux_arm64/jax.requirements.txt
# JAX requirements, passed into container by defining the ARG # REQUIREMENTS_FILE=jax.requirements.txt setuptools wheel cloudpickle colorama>=0.4.4 matplotlib pillow>=9.1.0 rich absl-py portpicker six opt-einsum auditwheel typing_extensions importlib_metadata>=4.6 numpy==1.26.0;python_version=="3.12" numpy==1.23.4;python_version=="3.11" numpy==1.22.4;python_version<"3.11" scipy==1.11.2;python_version=="3.12"
Plain Text - Registered: Tue May 07 12:40:20 GMT 2024 - Last Modified: Tue Jan 30 20:02:17 GMT 2024 - 570 bytes - Viewed (0) -
tensorflow/c/eager/immediate_execution_context.h
// Tensor creation functions virtual AbstractTensorInterface* CreateTensor( DataType dtype, absl::Span<const int64_t> dim_sizes) = 0; typedef void (*MemoryReleaser)(void* data, size_t len, void* arg); // Create a tensor instance from the given data buffer and description. // `memory_releaser` will be called on destruction, and it's responsible for // cleaning up the underlying buffer.
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jul 06 08:34:00 GMT 2023 - 12.3K bytes - Viewed (0) -
tensorflow/c/c_api_test.cc
EXPECT_EQ(TF_CANCELLED, TF_GetCode(s)); EXPECT_EQ(string("cancel"), TF_Message(s)); TF_DeleteStatus(s); } void Deallocator(void* data, size_t, void* arg) { tensorflow::cpu_allocator()->DeallocateRaw(data); *reinterpret_cast<bool*>(arg) = true; } TEST(CAPI, Tensor) { const int num_bytes = 6 * sizeof(float); float* values = reinterpret_cast<float*>(tensorflow::cpu_allocator()->AllocateRaw(
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 96.9K bytes - Viewed (3) -
ci/official/envs/linux_x86
TFCI_BUILD_PIP_PACKAGE_ARGS="--repo_env=WHEEL_NAME=tensorflow_cpu" TFCI_DOCKER_ENABLE=1 TFCI_DOCKER_IMAGE=tensorflow/build:2.16-python${TFCI_PYTHON_VERSION} TFCI_DOCKER_PULL_ENABLE=1 TFCI_DOCKER_REBUILD_ARGS="--build-arg PYTHON_VERSION=python$TFCI_PYTHON_VERSION --target=devel tensorflow/tools/tf_sig_build_dockerfiles" TFCI_INDEX_HTML_ENABLE=1 TFCI_LIB_SUFFIX="-cpu-linux-x86_64" TFCI_OUTPUT_DIR=build_output TFCI_WHL_AUDIT_ENABLE=1
Plain Text - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 23:12:40 GMT 2024 - 1.4K bytes - Viewed (0) -
tensorflow/c/experimental/next_pluggable_device/c_api.cc
TF_Status* status) { auto* cc_ctx = reinterpret_cast<tensorflow::OpKernelContext*>(ctx); const tensorflow::Tensor& arg_tensor = cc_ctx->input(index); absl::Status cc_status; if (arg_tensor.dtype() != tensorflow::DT_RESOURCE) { cc_status = absl::InvalidArgumentError( absl::StrCat("Trying to obtain resource handle from Input[", index,
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Tue Jan 09 00:52:04 GMT 2024 - 13.9K bytes - Viewed (1)