- Sort Score
- Result 10 results
- Languages All
Results 71 - 80 of 114 for var_handle_op (0.17 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/inlining.mlir
// inlined. func.func private @simple_callee_var() -> tensor<2xi32> { %cst = "tf.Const"() { value = dense<2> : tensor<2xi32> } : () -> tensor<2xi32> %0 = "tf.VarHandleOp"() {container = "c", shared_name = "v"} : () -> tensor<!tf_type.resource<tensor<2xi32>>> "tf.AssignVariableOp"(%0, %cst) {device = ""} : (tensor<!tf_type.resource<tensor<2xi32>>>, tensor<2xi32>) -> ()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 7.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/breakup-islands.mlir
// CHECK: } func.func @unknown_side_effecting_op(%arg0: tensor<32xf32>) -> () { tf_executor.graph { %island = tf_executor.island { %vh0 = "tf.VarHandleOp"() {container = "c", shared_name = "v0"} : () -> tensor<*x!tf_type.resource<tensor<32xf32>>> %vh1 = "tf.VarHandleOp"() {container = "c", shared_name = "v1"} : () -> tensor<*x!tf_type.resource<tensor<32xf32>>>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 31 08:59:10 UTC 2023 - 28.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/update_control_dependencies.mlir
func.func @unknown_side_effecting_op(%arg0: tensor<32xf32>) { tf_executor.graph { %outputs, %control = tf_executor.island wraps "tf.VarHandleOp"() {container = "c", shared_name = "v0"} : () -> tensor<*x!tf_type.resource<tensor<32xf32>>> %outputs_0, %control_1 = tf_executor.island wraps "tf.VarHandleOp"() {container = "c", shared_name = "v1"} : () -> tensor<*x!tf_type.resource<tensor<32xf32>>>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Nov 03 18:12:49 UTC 2023 - 25.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/mlir_import_options.h
bool unconditionally_use_set_output_shapes = false; // Apply default attributes from the op definition to the loaded op. bool add_default_attributes = true; // If set, promote tf.VarHandleOp to resource arguments for all functions. bool lift_variables = true; // Keeps the variables in initializers before lifting variables (when // `lift_variables == true`) or newly adding variable initialization patterns
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 20 13:19:26 UTC 2023 - 2.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/analysis/resource_value_typed_analyzer.cc
auto* operation = resource.getDefiningOp(); if (operation && isa<TF::VarHandleOp>(operation)) { mutable_variables_.insert(GetResourceKey(operation)); } } bool ResourceAnalyzer::IsPotentiallyWritten(Value resource) const { assert(IsResource(resource)); auto* operation = resource.getDefiningOp(); if (operation && isa<TF::VarHandleOp>(operation)) return mutable_variables_.contains(GetResourceKey(operation));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 15 09:04:13 UTC 2024 - 8K bytes - Viewed (0) -
tensorflow/c/experimental/saved_model/core/ops/variable_ops.h
#include "tensorflow/core/framework/tensor_shape.h" #include "tensorflow/core/framework/types.pb.h" #include "tensorflow/core/platform/status.h" namespace tensorflow { namespace internal { // Executes a VarHandleOp using `ctx`, and fills `handle` with the DT_RESOURCE // TensorHandle associated with the variable. This is equivalent to creating an // unitialized TF2 tf.Variable.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Aug 30 21:44:45 UTC 2022 - 3.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/freeze_global_tensors.cc
Operation *op = *latticeElement->getValue().ops.begin(); GlobalTensorOp globalTensor = llvm::dyn_cast<GlobalTensorOp>(op); if (!globalTensor) continue; // happens if the name is e.g. in a VarHandleOp. if (globalTensor.getIsMutable()) { freezeable[val] = false; continue; } freezeable[val] = true; // Verify users are supported kind.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.8K bytes - Viewed (0) -
tensorflow/c/experimental/ops/gen/cpp/golden/testing_ops.h.golden
//
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 16 19:04:03 UTC 2023 - 2.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_variable_runtime_reformatting.cc
// Creates the per-device variables that represent the formatting state of each // device. llvm::SmallVector<TF::VarHandleOp, 4> CreateStateVars( const llvm::SmallDenseMap<llvm::StringRef, llvm::SmallVector<StringRef, 4>>& devices, Location loc, RankedTensorType key_type, OpBuilder* builder) { llvm::SmallVector<TF::VarHandleOp, 4> state_vars; // TODO(b/148913020): Remove this constraint once model parallelism is // supported.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.9K bytes - Viewed (0) -
tensorflow/compiler/jit/build_xla_ops_pass_test.cc
result); } Node* MakeWrite(const Scope& scope, Output value_to_write, const string& id) { Output var_handle = ops::VarHandleOp(scope.WithOpName("Var_" + id), DT_FLOAT, TensorShape({})); ops::AssignVariableOp assign_op(scope.WithOpName("Assignee_" + id),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 12.2K bytes - Viewed (0)