- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for GetBodyAndConstantsAndResources (0.42 sec)
-
tensorflow/compiler/jit/force_xla_constants_on_host_pass.cc
NameAttrList function; TF_RETURN_IF_ERROR(NameAndAttrsFromFunctionCall(node->def(), &function)); // Force all constants to be on the host memory. TF_RETURN_IF_ERROR(GetBodyAndConstantsAndResources( flr, function, &fbody, &constant_arg_indices, &resource_arg_indices)); VLOG(3) << "Found constant arg indices: " << absl::StrJoin(constant_arg_indices, ", ");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 2.1K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_kernel_creator.cc
TF_RETURN_IF_ERROR(NameAndAttrsFromFunctionCall(node_def, &function)); const FunctionBody* fbody = nullptr; std::vector<int> constant_arg_indices; std::vector<int> resource_arg_indices; TF_RETURN_IF_ERROR(GetBodyAndConstantsAndResources( flr, function, &fbody, &constant_arg_indices, &resource_arg_indices)); MemoryTypeVector input_memory_types = GetInputMemoryTypes(fbody, constant_arg_indices, resource_arg_indices);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 31 22:24:01 UTC 2023 - 4.4K bytes - Viewed (0) -
tensorflow/compiler/jit/compilability_check_util.h
// `fbody` is owned by `flr`. // `constant_arg_indices` and `resource_arg_indices` should be empty vector. // They are sorted in ascending order on this function's return. Status GetBodyAndConstantsAndResources(FunctionLibraryRuntime* flr, const NameAttrList& function, const FunctionBody** fbody,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 14.9K bytes - Viewed (0) -
tensorflow/compiler/jit/get_compiler_ir.cc
CompilerArgSource compiler_arg_source) { const FunctionBody* fbody = nullptr; std::vector<int> constant_arg_indices; std::vector<int> resource_arg_indices; TF_RETURN_IF_ERROR(GetBodyAndConstantsAndResources( flr, function, &fbody, &constant_arg_indices, &resource_arg_indices)); if (dev == nullptr && !resource_arg_indices.empty()) { return absl::InternalError(
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/compilability_check_util.cc
return it != node.attr().end() && it->second.b(); } bool CanCreateXlaKernel(const NodeDef& node_def) { return HasBoolAttr(node_def, kXlaMustCompileAttr); } Status GetBodyAndConstantsAndResources(FunctionLibraryRuntime* flr, const NameAttrList& function, const FunctionBody** fbody,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 30.3K bytes - Viewed (0)