Search Options

Results per page
Sort
Preferred Languages
Advance

Results 71 - 80 of 142 for var_handle_op (0.19 sec)

  1. 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)
  2. 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)
  3. tensorflow/c/eager/parallel_device/parallel_device_lib_test.cc

          "/job:localhost/replica:0/task:0/device:CPU:1"};
      ParallelDevice parallel_device(std::move(devices));
      std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> handle_op(
          TFE_NewOp(context.get(), "VarHandleOp", status.get()), TFE_DeleteOp);
      ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get());
      TFE_OpSetAttrType(handle_op.get(), "dtype", TF_FLOAT);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jul 08 23:47:35 UTC 2021
    - 15.3K bytes
    - Viewed (0)
  4. 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)
  5. tensorflow/cc/tools/freeze_saved_model_test.cc

        if (use_resource) {
          Output var =
              ops::VarHandleOp(scope.WithOpName("var"), DataType::DT_FLOAT, {});
          read_var = ops::ReadVariableOp(
              scope.WithOpName("var/Read/ReadVariableOp"), var, DataType::DT_FLOAT);
          auto assign = ops::AssignVariableOp(scope.WithOpName("assign"), var, a);
          Output var_1 =
              ops::VarHandleOp(scope.WithOpName("var_1"), DataType::DT_FLOAT, {});
          Output read_var_1 =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 07 13:30:31 UTC 2022
    - 21.7K bytes
    - Viewed (0)
  6. 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)
  7. 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)
  8. 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)
  9. tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model_initialize_variables_in_session_init_fail.mlir

      attributes {tf.entry_function = {control_outputs = "", inputs = "", outputs = "dense_2/Add:0"}, tf_saved_model.exported_names = ["serving_default"]} {
        %0 = "tf.VarHandleOp"() {_class = ["loc:@dense/kernel"], allowed_devices = [], container = "", device = "/job:worker/replica:0/task:1/device:CPU:0", shared_name = "var1"} : () -> tensor<!tf_type.resource<tensor<100x50xf32>>>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Mar 24 05:47:26 UTC 2022
    - 994 bytes
    - Viewed (0)
  10. tensorflow/c/experimental/ops/update_cpp_ops.sh

      SqrtGrad \
      Log1p
    
    ${generate} \
      --category=nn \
      SparseSoftmaxCrossEntropyWithLogits \
      ReluGrad \
      Relu \
      BiasAdd \
      BiasAddGrad
    
    ${generate} \
      --category=resource_variable \
      VarHandleOp \
      ReadVariableOp \
      AssignVariableOp \
      DestroyResourceOp
    
    ${generate} \
      --category=io \
      RestoreV2 \
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 17 17:54:34 UTC 2022
    - 1.6K bytes
    - Viewed (0)
Back to top