- Sort Score
- Result 10 results
- Languages All
Results 111 - 120 of 142 for varhandle_op (0.22 sec)
-
tensorflow/compiler/mlir/tfrt/tests/mlrt/tf_to_mlrt.mlir
// Test lowering of IfrtLoadVariableOp // CHECK-LABEL: func @ifrt_load_variable_test func.func @ifrt_load_variable_test() -> () { // CHECK: [[HANDLE:%.*]] = tf_mlrt.executeop() // CHECK-SAME: VarHandleOp %0 = "tf.VarHandleOp"() {__op_key = 1: i32, device = "/device:CPU:0", container = "", shared_name = "variable"} : () -> tensor<!tf_type.resource<tensor<1x3xf32>>> // CHECK-NEXT: "tf_mlrt.ifrt_load_variable"([[HANDLE]])
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 20:44:15 UTC 2024 - 24.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/analyze_variables.cc
// Returns true if 'op' is TF op that accepts resource type, but is // supported by TFLite. bool IsSupportedTFLiteResourceOp(Operation* op) { return llvm::isa<TF::ReadVariableOp, TF::AssignVariableOp, TF::VarHandleOp, TF::LookupTableFindV2Op, TF::LookupTableImportV2Op, TF::LookupTableSizeV2Op>(op); } // Returns true if 'op' is TF/TFLite control flow op that can accept resource
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tpu_cluster_formation.mlir
func.func @resource_before_cluster() { // CHECK-NEXT: [[CONST:%.*]] = "tf.Const" %0 = "tf.Const"() {value = dense<1.000000e+00> : tensor<f32>} : () -> tensor<f32> // CHECK-NEXT: [[UNUSED_RESOURCE:%.*]] = "tf.VarHandleOp" // CHECK-NEXT: "tf.AssignAddVariableOp"([[UNUSED_RESOURCE]], [[CONST]]) // CHECK: "tf_device.cluster" // CHECK-NEXT: "tf.NoOp" // CHECK-NEXT: tf_device.return
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 22:03:30 UTC 2024 - 53.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/mlrt/parallelization.mlir
// CHECK-LABEL: func private @main_stream_{{[0-9]*}} // CHECK-SAME: ([[CONTROL_FUTURE:%.*]]: !mlrt.future, [[PROMISE:%.*]]: !mlrt.promise, [[PROMISE_HANDLE:%.*]]: !mlrt.promise) // CHECK: [[HANDLE:%.*]] = "tf.VarHandleOp" // CHECK-NEXT: "tf_mlrt.tf_promise"([[PROMISE_HANDLE]], [[HANDLE]]) // CHECK: mlrt.await_control [[CONTROL_FUTURE]] // CHECK-NEXT: [[V:%.*]] = "tf.ReadVariableOp"([[HANDLE]])
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 08 22:07:30 UTC 2023 - 15K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/merge_initializer_function_ops_to_main.mlir
%out_0, %ctl_0 = tf_executor.island wraps "tf.Const"() {value = dense<"var_0"> : tensor<1x!tf_type.string>} : () -> tensor<1x!tf_type.string> %out_1, %ctl_1 = tf_executor.island wraps "tf.VarHandleOp"() {container = "", shared_name = "var_0", device = "/device:CPU:0"} : () -> tensor<!tf_type.resource<tensor<2xf32>>>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 31 08:49:35 UTC 2023 - 29.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/device_canonicalize.mlir
func.func @cluster_result_for_resource_update(%arg0 : tensor<!tf_type.string>, %arg1 : tensor<!tf_type.string>) -> (tensor<!tf_type.string>, tensor<!tf_type.string>) { %resource = "tf.VarHandleOp"() {container = "c", shared_name = "v"} : () -> tensor<*x!tf_type.resource<tensor<*x!tf_type.string>>> // CHECK: %[[RESULT:.*]] = "tf_device.cluster" %0:2 = "tf_device.cluster"() ({ // CHECK: "tf.Equal"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Nov 04 14:07:37 UTC 2022 - 6.4K bytes - Viewed (0) -
tensorflow/c/eager/custom_device_test.cc
// Create a variable handle placed on the custom device. std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> op( TFE_NewOp(context.get(), "VarHandleOp", status.get()), TFE_DeleteOp); ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get()); TFE_OpSetAttrType(op.get(), "dtype", TF_FLOAT); TFE_OpSetAttrShape(op.get(), "shape", {}, 0, status.get());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 27 23:39:24 UTC 2020 - 18.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.td
let constructor = "TF::CreateHoistLoopInvariantPass()"; } def LocalizeVarHandlesPass : Pass<"tf-localize-var-handles", "ModuleOp"> { let summary = "Creates VarHandleOps next to the operations that use them."; let description = [{ Creates VarHandleOps right next to the operations that use them, one per operation. This is useful for transformations that only end up with a few small
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 99.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_savedmodel_passes.td
for those which come from a "bound input" parameter. For the latter, said parameter will not be removed, unlike LowerGlobalsToMlProgramPass. So this will lower e.g. %0 = "tf.VarHandleOp" {name = "foo"} %1 = "tf.ReadVariableOp"(%0) to %0 = ml_program.global_load @vars.foo }]; let constructor = "::mlir::tf_saved_model::CreateLowerVariableOpsToMlProgramPass()";
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jun 09 19:11:34 UTC 2023 - 7.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lower_globals_to_ml_program.cc
if (auto global = tf_saved_model::LookupBoundInputOfType< tf_saved_model::GlobalTensorOp>(func, i, syms)) { OpBuilder builder(func.getBody()); auto dummy = builder.create<TF::VarHandleOp>( global.getLoc(), func.getArgument(i).getType(), "dummy", "dummy"); func.getArgument(i).replaceAllUsesWith(dummy.getResult()); argsToErase.set(i); } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.6K bytes - Viewed (0)