- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 58 for varhandle_op (0.17 sec)
-
tensorflow/compiler/mlir/tfrt/tests/ifrt/tf_restore_pruning.mlir
// CHECK-NOT: tf.RestoreV2 %0 = "tf.RestoreV2"(%cst, %cst_1, %cst_0): (tensor<!tf_type.string>, tensor<1x!tf_type.string>, tensor<1x!tf_type.string>) -> tensor<3x1xf32> %1 = "tf.VarHandleOp"() <{container = "", shared_name = "y"}> : () -> tensor<!tf_type.resource<tensor<3x1xf32>>> return } // CHECK-LABEL: func.func @used_restore_remains func.func @used_restore_remains() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 25 22:02:06 UTC 2024 - 1.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/merge_save_function_ops_to_main.mlir
module attributes {tf_saved_model.semantics} { func.func private @tf_quant__save(%arg: tensor<!tf_type.string>) -> () { tf_executor.graph { %out, %ctl = tf_executor.island wraps "tf.VarHandleOp"() {shared_name = "var_0"} : () -> tensor<!tf_type.resource<tensor<2xf32>>> %out_0, %ctl_0 = tf_executor.island wraps "tf.ReadVariableOp"(%out) : (tensor<!tf_type.resource<tensor<2xf32>>>) -> tensor<2xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 7.7K bytes - Viewed (0) -
tensorflow/c/experimental/ops/gen/cpp/golden/testing_ops.cc.golden
int num_retvals = 0; std::vector<AbstractTensorHandle*> dummy_outputs; return op_ptr->Execute(absl::MakeSpan(dummy_outputs), &num_retvals); } // Op: VarHandleOp() // Summary: // // Description:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 16 19:04:03 UTC 2023 - 6.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/mlprogram.mlir
%0 = tf_executor.graph { %outputs_4, %control_5 = tf_executor.island wraps "tf.VarHandleOp"() {container = "", shared_name = "Variable"} : () -> tensor<!tf_type.resource<tensor<!tf_type.string>>> %outputs_10, %control_11 = tf_executor.island wraps "tf.VarHandleOp"() {container = "", shared_name = "Variable_1"} : () -> tensor<!tf_type.resource<tensor<!tf_type.string>>>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 19:27:16 UTC 2024 - 7.4K bytes - Viewed (0) -
tensorflow/compiler/aot/aot_only_var_handle_op.cc
#include "tensorflow/compiler/tf2xla/xla_op_registry.h" #include "tensorflow/core/framework/shape_inference.h" namespace tensorflow { namespace { // Implementation of varhandle that binds a VarHandleOp to an XlaResource of the // same name. It is not safe to use this op in a JIT context. class XlaAotOnlyVarHandleOp : public XlaOpKernel { public: explicit XlaAotOnlyVarHandleOp(OpKernelConstruction* c);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 28 09:57:04 UTC 2024 - 3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model/import_restore_v1.py
# CHECK: [[v1:%.*]] = "tf.Identity"([[v0]]) # CHECK: [[handle:%.*]] = "tf.VarHandleOp" # CHECK-SAME: shared_name = [[shared_name:".*"]] # CHECK: "tf.AssignVariableOp"([[handle]], [[v1]]) # CHECK: func {{@[a-zA-Z_0-9]+}}( # CHECK-SAME: tf_saved_model.exported_names = ["key"] # CHECK: tf.VarHandleOp # CHECK-SAME: shared_name = [[shared_name]] def Test():
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 31 08:49:35 UTC 2023 - 2.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/basic.mlir
// CHECK-NOT: "tf_saved_model.session_initializer" "tf_saved_model.session_initializer"() { initializers = [@func_init] } : () -> () // CHECK-LABEL: _tfrt_resource_init // CHECK: tf.VarHandleOp // CHECK: tf.ReadVariableOp // CHECK: tfrt_fallback_async.set_resource // CHECK-SAME: {device = "/device:CPU:0", index = 0 : i64} // CHECK-LABEL: func @init // CHECK-SAME: {tfrt.cost_threshold = 1 : i64}
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 00:18:59 UTC 2024 - 3.9K bytes - Viewed (0) -
tensorflow/compiler/jit/tests/keras_imagenet_main_graph_mode.golden_summary
Clustered nodes: 2178 Unclustered nodes: 446 Number of clusters: 1 unclustered size 446 AssignAddVariableOp 2 Const 2 DivNoNan 1 Identity 1 NoOp 1 ReadVariableOp 2 VarHandleOp 435 _Retval 2 cluster 0 size 2178 Add 17 AddN 72 ArgMax 1 AssignAddVariableOp 1 AssignSubVariableOp 106 BiasAdd 1 BiasAddGrad 1 Cast 3 Const 357 Conv2D 53 Conv2DBackpropFilter 53 Conv2DBackpropInput 52
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jan 06 10:38:14 UTC 2023 - 740 bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/tf_to_corert_pipeline_refvar.mlir
// CHECK-NEXT: [[o_chain:%.*]], [[o:%.*]] = tfrt_fallback_async.executeop.seq([[in_chain]]) key(0) cost({{.*}}) device("/job:localhost/replica:0/task:0/device:CPU:0") "tf.VarHandleOp"() // CHECK-NEXT: [[o_chain_0:%.*]], [[o1:%.*]] = tfrt_fallback_async.executeop.seq([[in_chain]]) key(1) cost({{.*}}) device("/job:localhost/replica:0/task:0/device:CPU:0") "tf.ReadVariableOp"([[o]]) {dtype = f32} : 1
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 00:18:59 UTC 2024 - 1.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/saved_model/testdata/xla_launch_xla_reduce_window.mlir
func.return %0 : tensor<10xf32> } func.func @main(%arg0: tensor<7xf32>) -> tensor<10xf32> attributes {tf.entry_function = {control_outputs = "", inputs = "input:0", outputs = "output:0"}} { %0 = "tf.VarHandleOp"() {device = "/device:CPU:0", container = "", shared_name = "variable"} : () -> tensor<!tf_type.resource<tensor<f32>>> %1 = "tf.ReadVariableOp"(%0) {device = "/device:CPU:0"} : (tensor<!tf_type.resource<tensor<f32>>>) -> tensor<f32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Aug 14 15:35:49 UTC 2023 - 1.6K bytes - Viewed (0)