- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 78 for var_handle_op (0.56 sec)
-
tensorflow/compiler/mlir/lite/tests/legalize-tf-variables.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 7.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model_lift_variables_invalid_session.mlir
%0 = "tf.VarHandleOp"() {_class = ["loc:@dense/kernel"], allowed_devices = [], container = "", device = "", shared_name = "dense/kernel"} : () -> tensor<!tf_type.resource<tensor<100x50xf32>>> %1 = "tf.ReadVariableOp"(%0) {device = ""} : (tensor<!tf_type.resource<tensor<100x50xf32>>>) -> tensor<100x50xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 24 05:47:26 UTC 2022 - 2.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/side_effects.mlir
// CHECK-SAME: ([[in_chain:%.*]]: !tfrt.chain) -> !tfrt.chain func.func @assign_variable() { // CHECK: [[ch1:%.*]], %results = tfrt_fallback_async.executeop.seq([[in_chain]]) key(0) cost({{.*}}) device("/device:CPU:0") "tf.VarHandleOp" // CHECK-NEXT: [[ch2:%.*]] = tfrt_fallback_async.executeop.seq([[in_chain]]) key(1) cost({{.*}}) device("/device:CPU:0") "tf.AssignVariableOp" // CHECK-NEXT: [[out_ch:%.*]] = tfrt.merge.chains [[ch1]], [[ch2]]
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 28 12:06:33 UTC 2022 - 1008 bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/integration/node_expansion_test.py
# Regression test for an issue where VarHandleOp wasn't being properly # imported into MLIR for "no-op" node expansion. def testVarHandleOp(self): x = constant_op.constant([[1.0, 2.0], [3.0, 4.0], [5.0, 6.0]]) # Note: we purposely make multiple calls to VarHandleOp to exercise the # cached kernal lookup path that was exhibiting the VarHandleOp import # issue.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Sep 28 21:37:05 UTC 2021 - 3.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/analyze-variables.mlir
// RUN: tf-opt %s -split-input-file -tfl-analyze-variables-pass --cse | FileCheck %s // CHECK: module attributes {tfl._legalize_tfl_variables = true} module { func.func @f() -> tensor<*xi32> { %0 = "tf.VarHandleOp"() {container = "c", shared_name = "v"} : () -> tensor<*x!tf_type.resource<tensor<*xi32>>> %2 = "tf.ReadVariableOp"(%0) {dtype = i32} : (tensor<*x!tf_type.resource<tensor<*xi32>>>) -> tensor<*xi32> func.return %2 : tensor<*xi32> } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 09 11:49:28 UTC 2022 - 5.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/localize_var_handles.cc
} else if (auto next = llvm::dyn_cast<TF::IteratorGetNextOp>(op)) { resource = next.getIterator(); } if (llvm::dyn_cast_or_null<TF::VarHandleOp>(resource.getDefiningOp())) { return; // We're already directly after a VarHandleOp. } const TF::ResourceDataflowState* state = solver.lookupState<TF::ResourceDataflowState>(resource); if (!state) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 06 23:53:00 UTC 2024 - 5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/convert_ref_variables.mlir
// Test the basic cases where all uses of a ref variable can be converted. // CHECK-LABEL: @init func.func @init() { // CHECK-NOT: tf.VariableV2 // CHECK-NOT: tf.Assign // CHECK: [[handle:%.*]] = "tf.VarHandleOp" // CHECK-SAME: shared_name = "x" // CHECK: "tf.AssignVariableOp"([[handle]], {{%.*}}) %0 = "tf.VariableV2"() {container = "", shape = #tf_type.shape<>, shared_name = "x"} : () -> tensor<!tf_type.int32ref>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 28 12:06:33 UTC 2022 - 4.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/mlrt/rewrite_ifrt_load_variable.mlir
// Variable is used by both CPU and TPU // // CHECK-LABEL: func @serving_default(%arg0: tensor<1x3xf32>) -> tensor<1x1xf32> // CHECK-NEXT: [[HANDLE:%.*]] = "tf.VarHandleOp"() // CHECK-NEXT: [[ARRAYKEY:%.*]], [[FURTURE:%.*]] = "tf_mlrt.tf_ifrt_load_variable"([[HANDLE]]) // CHECK-SAME: <{used_by_host = true}> : (tensor<!tf_type.resource<tensor<3x1xf32>>>) -> (tensor<!tf_type.string>, !mlrt.future)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 21:35:32 UTC 2024 - 1.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/resource_inlining.mlir
// CHECK-LABEL: func @main func.func @main() -> tensor<f32> { // CHECK-NEXT: %[[VAR:.*]] = "tf.VarHandleOp" // CHECK-NEXT: %[[READ_VAR:.*]] = "tf.ReadVariableOp"(%[[VAR]]) // CHECK-NEXT: return %[[READ_VAR]] // CHECK-NOT: "tf.Cast"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 24 05:47:26 UTC 2022 - 1.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/graph-function-resource-args.pbtxt
# RUN: tf-mlir-translate -graphdef-to-mlir -tf-enable-shape-inference-on-import=false %s -tf-output-arrays=func_call -o - | FileCheck %s node { name: "x" op: "VarHandleOp" device: "/CPU:0" attr { key: "container" value { s: "a" } } attr { key: "dtype" value { type: DT_INT64 } } attr { key: "shape" value { shape { } } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 1.8K bytes - Viewed (0)