- Sort Score
- Result 10 results
- Languages All
Results 1 - 9 of 9 for variable_info (0.19 sec)
-
tensorflow/compiler/jit/xla_launch_util_gpu_test.cc
#include "tensorflow/compiler/jit/flags.h" #include "tensorflow/compiler/jit/pjrt_device_compiler_client.h" #include "tensorflow/compiler/jit/pjrt_device_context.h" #include "tensorflow/compiler/jit/variable_info.h" #include "tensorflow/compiler/jit/variable_info_util.h" #include "tensorflow/compiler/jit/xla_launch_util.h" #include "tensorflow/compiler/tf2xla/xla_op_registry.h" #include "xla/pjrt/pjrt_client.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 10K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_compile_on_demand_op.cc
absl::StatusOr<std::vector<VariableInfo>> variable_infos = GatherVariableInfo(ctx, *result, 0); TF_RETURN_IF_ERROR(variable_infos.status()); TF_RETURN_IF_ERROR(LockVariables(absl::MakeSpan(*variable_infos))); TF_RETURN_IF_ERROR(launch_context.PopulateOutputs( ctx, result, execution_output.ConsumeResult(), /*missing_ctx_input_prefix=*/0, absl::MakeSpan(*variable_infos), input_output_alias, snapshot_ptrs));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 29 08:39:39 UTC 2024 - 13.4K bytes - Viewed (0) -
tensorflow/c/experimental/next_pluggable_device/c_api.cc
// operate on TF_VariableInfo (such as allocate temp tensor for the `var` held // by the underlying tensorflow::VariableInfo. TF_VariableInfo(int index, const std::string& name, tensorflow::Var* var) { var_info = tensorflow::VariableInfo{index, name, var}; } tensorflow::VariableInfo var_info{0, "", nullptr}; }; TF_VariableInfo* TF_CreateVariableInfoFromContext(TF_OpKernelContext* ctx,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 05:48:24 UTC 2024 - 13.9K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_launch_util.h
Status PopulateOutputs( OpKernelContext* ctx, const XlaCompiler::CompilationResult* compilation_result, xla::ScopedShapedBuffer output, int missing_ctx_input_prefix, absl::Span<VariableInfo> variable_infos, const xla::HloInputOutputAliasConfig& input_output_alias, const std::map<int, const Tensor*>& resource_vars); private: xla::LocalClient* client_;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 11.8K bytes - Viewed (0) -
tensorflow/compiler/jit/get_compiler_ir.cc
inputs, resource_arg_indices, &variable_infos)); TF_RETURN_IF_ERROR(LockVariables(absl::MakeSpan(variable_infos))); } absl::StatusOr<std::vector<XlaCompiler::Argument>> args; if (compiler_arg_source == CompilerArgSource::TENSOR_SPEC) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 06:59:07 UTC 2024 - 19K bytes - Viewed (0) -
tensorflow/compiler/jit/kernels/xla_ops.cc
std::pair<std::vector<XlaCompiler::Argument>, ResourceVarsSnapshot> result; std::vector<VariableInfo> variable_infos; TF_RETURN_IF_ERROR( GetVariableInfosFromInputs(ctx->resource_manager(), ctx->device(), inputs, variable_indices, &variable_infos)); TF_RETURN_IF_ERROR(LockVariables(absl::MakeSpan(variable_infos))); TF_RETURN_IF_ERROR(SnapshotResourceVariables(ctx, variable_indices,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 22:46:36 UTC 2024 - 41.4K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_launch_util.cc
ctx->set_output(i, output_tensor); ++output_num; } } // input_index -> index into variable_infos. absl::flat_hash_map<int, int> variable_info_lookup; for (int i = 0; i < variable_infos.size(); i++) { variable_info_lookup.emplace(variable_infos[i].index(), i); } // Apply variable updates, if any. for (int i = 0, end = compilation_result->resource_updates.size(); i < end;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 00:36:08 UTC 2024 - 40.4K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_launch_util_test.cc
#include "tensorflow/compiler/jit/device_compiler.h" #include "tensorflow/compiler/jit/flags.h" #include "tensorflow/compiler/jit/pjrt_device_compiler_client.h" #include "tensorflow/compiler/jit/variable_info.h" #include "tensorflow/compiler/jit/variable_info_util.h" #include "tensorflow/compiler/tf2xla/xla_op_registry.h" #include "xla/pjrt/pjrt_client.h" #include "xla/pjrt/tfrt_cpu_pjrt_client.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 28.8K bytes - Viewed (0) -
pkg/env/var.go
// specializedVar represents a var that can Get/Lookup type specializedVar[T any] interface { Lookup() (T, bool) Get() T } // VariableInfo provides generic information about a variable. All Variables implement this interface. // This is largely to workaround lack of covariance in Go. type VariableInfo interface { GetName() string IsSet() bool
Registered: Fri Jun 14 15:00:06 UTC 2024 - Last Modified: Tue May 23 17:08:31 UTC 2023 - 11.8K bytes - Viewed (0)