- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for GetVariableInfosFromInputs (0.2 sec)
-
tensorflow/compiler/jit/variable_info_util.cc
#include "tsl/platform/status.h" namespace tensorflow { Status GetVariableInfosFromInputs(ResourceMgr* rm, DeviceBase* dev, absl::Span<const Tensor* const> inputs, absl::Span<const int> variable_indices, std::vector<VariableInfo>* result) { return GetVariableInfosFromInputs(rm, dev, inputs, variable_indices, nullptr,
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/variable_info_util.h
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Feb 14 21:57:02 UTC 2023 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_launch_util_test.cc
std::vector<int> variables_indices = GetResourceVariableIndicesFromContext(context_.get()); std::vector<VariableInfo> variables; variables.reserve(variables_indices.size()); TF_ASSERT_OK(GetVariableInfosFromInputs(context_->resource_manager(), context_->device(), inputs, variables_indices, &variables));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 28.8K 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)));
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_compile_on_demand_op.cc
const std::vector<int>& variable_indices, std::vector<VariableInfo>* variables, std::vector<XlaCompiler::Argument>* args) { variables->clear(); args->clear(); TF_RETURN_IF_ERROR(GetVariableInfosFromInputs(ctx.resource_manager(), ctx.device(), inputs, variable_indices, variables));
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/compiler/jit/get_compiler_ir.cc
inputs[i + offset] = &inputs_storage.back(); } else { inputs[i + offset] = t; } } if (dev != nullptr) { TF_RETURN_IF_ERROR(GetVariableInfosFromInputs(dev->resource_manager(), dev, inputs, resource_arg_indices, &variable_infos));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 06:59:07 UTC 2024 - 19K bytes - Viewed (0)