- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 14 for variable_info (0.33 sec)
-
tensorflow/compiler/jit/variable_info.cc
#include "tsl/platform/status.h" namespace tensorflow { VariableInfo::VariableInfo( int index, absl::string_view name, Var* var, const std::optional<ManagedStackTrace>& definition_stack_trace) : index_(index), name_(name), var_(var), definition_stack_trace_(definition_stack_trace) {} VariableInfo::VariableInfo(VariableInfo&& other) : index_(other.index_), var_(other.var_),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 2.2K bytes - Viewed (0) -
tensorflow/compiler/jit/BUILD
"defs.h", ], visibility = [":friends"], ) # Internal targets below this point. cc_library( name = "variable_info", srcs = ["variable_info.cc"], hdrs = ["variable_info.h"], visibility = [ ":internal", # We reuse VariableInfo in TFRT's implementation of TpuExecuteOp. "//learning/brain/tfrt/tf_tpu:__pkg__", "//learning/brain/tfrt/tpu_plugin:__pkg__",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 00:41:19 UTC 2024 - 61.5K bytes - Viewed (0) -
tensorflow/compiler/jit/variable_info_util.cc
*ptr = new Var(DT_INVALID); return absl::OkStatus(); })); VariableInfo& variable_info = result->emplace_back( var_idx, handle.name(), variable, handle.definition_stack_trace()); if (variables_updated != nullptr && variables_updated->find(var_idx) == variables_updated->end()) { variable_info.set_read_only(); } } return absl::OkStatus(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 7K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_compile_on_demand_op.h
#define TENSORFLOW_COMPILER_JIT_XLA_COMPILE_ON_DEMAND_OP_H_ #include <vector> #include "tensorflow/compiler/jit/device_compilation_profiler.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/jit/xla_platform_info.h" #include "tensorflow/compiler/tf2xla/xla_compiler.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 3.2K bytes - Viewed (0) -
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)