- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for resource_updates (0.13 sec)
-
tensorflow/compiler/jit/xla_launch_util.cc
int missing_ctx_input_prefix) { std::vector<VariableInfo> out; out.reserve(compilation_result.resource_updates.size()); for (int i = 0; i < compilation_result.resource_updates.size(); ++i) { const XlaCompiler::ResourceUpdate& write = compilation_result.resource_updates[i]; int actual_input_index = write.input_index - missing_ctx_input_prefix;
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/mlir/tf2xla/api/v1/compile_mlir_util.cc
// Add resource write. resource_updates->emplace_back(); XlaResourceUpdate& resource_update = resource_updates->back(); resource_update.input_index = it->getSecond(); resource_update.modified = true; TF_RETURN_IF_ERROR(ConvertToDataType(tensor_type, &resource_update.type)); TF_RETURN_IF_ERROR(XLAShapeToTensorShape(shape, &resource_update.shape)); continue; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 17:24:39 UTC 2024 - 45.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/tf_xla_mlir_translate.cc
output << '\n'; } for (const auto& resource_update : compilation_result.resource_updates) { output << "// ResourceUpdate input_index=" << resource_update.input_index << " type=" << DataTypeString(resource_update.type) << " shape=(" << absl::StrJoin(resource_update.shape.dim_sizes(), " ") << ')'; if (resource_update.modified) output << " modified"; output << '\n'; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 18.8K bytes - Viewed (0) -
tensorflow/compiler/jit/kernels/xla_ops.cc
std::vector<VariableInfo>* variable_infos) { std::set<int> variables_updated; for (const auto& resource_update : compilation_result.resource_updates) { if (resource_update.modified) { variables_updated.insert(resource_update.input_index); } } return GetVariableInfosFromInputs(ctx->resource_manager(), ctx->device(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 22:46:36 UTC 2024 - 41.4K bytes - Viewed (0)