- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for session_init_funcs (0.33 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/initialize_variables_in_session_init.cc
func::FuncOp session_init_func, OpBuilder builder) { absl::StatusOr<ElementsAttr> tensor_attr_or = tensorflow::ConvertTensor(*tensor, &builder); assert(tensor_attr_or.ok() && "Expect valid tensor"); ElementsAttr tensor_attr = tensor_attr_or.value(); builder.setInsertionPointToStart(&session_init_func.getBlocks().front()); auto var_handle_op_in_init = var_handle_op->clone();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 23 09:05:47 UTC 2024 - 7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_save_op.cc
// `tf.VarHandleOp`s that are initialized by these `tf.AssignVariableOp`s. SmallVector<TF::VarHandleOp> CollectVariableOps( func::FuncOp session_init_func) { SmallVector<TF::VarHandleOp> var_handle_ops{}; for (auto assign_variable_op : llvm::make_early_inc_range( session_init_func.getOps<TF::AssignVariableOp>())) { Value resource_operand = assign_variable_op.getOperand(0); auto var_handle_op =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 05:52:39 UTC 2024 - 9.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/analysis/resource_value_typed_analyzer.cc
} } // namespace ResourceAnalyzer::ResourceAnalyzer(ModuleOp module, bool skip_session_init) { auto session_init_func = GetSessionInitializerFunc(module); for (auto func : module.getOps<func::FuncOp>()) { if (skip_session_init && func == session_init_func) continue; (void)AnalyzeRegion(func.getRegion()); } } void ResourceAnalyzer::SetPotentiallyWritten(Value resource) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 15 09:04:13 UTC 2024 - 8K bytes - Viewed (0)