Search Options

Results per page
Sort
Preferred Languages
Advance

Results 21 - 30 of 33 for AddN (0.11 sec)

  1. tensorflow/compiler/mlir/tensorflow/tests/lower_tf.mlir

      // CHECK: tf.AddN
      %0 = "tf.AddN"(%arg0, %arg1, %arg2) : (tensor<!tf_type.variant<tensor<2xf32>>>, tensor<!tf_type.variant<tensor<2xf32>>>, tensor<!tf_type.variant<tensor<2xf32>>>) -> tensor<!tf_type.variant<tensor<2xf32>>>
      func.return %0 : tensor<!tf_type.variant<tensor<2xf32>>>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Jan 05 18:35:42 UTC 2024
    - 92K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tf2xla/tests/legalize-tf-with-tf2xla-hlo-importer.mlir

      }
    
      // CHECK-LABEL: unsupported_dtype
      func.func @unsupported_dtype(%arg0: tensor<2x!tf_type.variant>) -> tensor<2x!tf_type.variant> {
        // CHECK: tf.AddN
        // expected-remark@+1 {{skipping legalization due to unsupported type 'tensor<2x!tf_type.variant>'}}
        %0 = "tf.AddN"(%arg0, %arg0) : (tensor<2x!tf_type.variant>, tensor<2x!tf_type.variant>) -> tensor<2x!tf_type.variant>
    
        func.return %0 : tensor<2x!tf_type.variant>
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Apr 06 15:32:52 UTC 2024
    - 38.6K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tfrt/tests/mlrt/parallelization.mlir

      %b2 = "tf.Sub"(%b1, %b) : (tensor<i32>, tensor<i32>) -> tensor<i32>
      %b3 = "tf.Sub"(%b2, %b) : (tensor<i32>, tensor<i32>) -> tensor<i32>
    
      %d = "tf.AddN"(%a3, %b3, %c) : (tensor<i32>, tensor<i32>, tensor<i32>) -> tensor<i32>
      func.return %d : tensor<i32>
    }
    
    // -----
    
    // Test that constants are copied instead of using promise/await.
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 08 22:07:30 UTC 2023
    - 15K bytes
    - Viewed (0)
  4. tensorflow/cc/framework/gradients.cc

        // Just one backprop edge.
        *grad = grads_to_keep[0];
      } else {
        // Otherwise, adds backprop-ed gradients.
        // TODO(andydavis) Use a better accumulator here.
        *grad = ops::AddN(scope_, grads_to_keep);
      }
    
      return absl::OkStatus();
    }
    
    bool SymbolicGradientBuilder::IsPrimitiveOpWithNoGrad(const string& opname) {
      ops::GradFunc grad_fn;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Apr 13 05:57:22 UTC 2024
    - 22K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tfr/python/tfr_gen_test.py

    @composite.Composite('TestInputNOp')
    def _tfr_control_flow_tensor_list_size(ins):
      n = len(ins)
      if n == 0:
        return array_ops.Const(value=[[0, 1], [2, 3]], dtype=dtypes.int64)
      else:
        return math_ops.AddN(ins)
    
    
    #--- test fn for tf ops ---
    
    
    @composite.Composite('TestComplexTFOp')
    def _tfr_tf_ops_complex(lhs, rhs):
      left_padding, _ = array_ops.SplitV(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Oct 13 16:33:28 UTC 2021
    - 28.8K bytes
    - Viewed (0)
  6. tensorflow/c/c_api_test.cc

      ASSERT_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s);
    
      // Test TF_Operation*() query functions.
      EXPECT_EQ(string("add"), string(TF_OperationName(add)));
      EXPECT_EQ(string("AddN"), string(TF_OperationOpType(add)));
      EXPECT_EQ(string(""), string(TF_OperationDevice(add)));
      EXPECT_EQ(1, TF_OperationNumOutputs(add));
      EXPECT_EQ(TF_INT32, TF_OperationOutputType(TF_Output{add, 0}));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 15 03:35:10 UTC 2024
    - 96.9K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/lite/tests/legalize-tf.mlir

    }
    
    func.func @addN(%arg0: tensor<2x3xi32>, %arg1: tensor<2x3xi32>, %arg2: tensor<2x3xi32>) -> tensor<2x3xi32> {
      %0 = "tf.AddN"(%arg0, %arg1, %arg2) : (tensor<2x3xi32>, tensor<2x3xi32>, tensor<2x3xi32>) -> tensor<2x3xi32>
      func.return %0 : tensor<2x3xi32>
    
    // CHECK-LABEL: addN
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 05 01:54:33 UTC 2024
    - 153.4K bytes
    - Viewed (0)
  8. tensorflow/c/c_api_function_test.cc

      TF_Operation* add = Add(feed1, feed2, func_graph_, s_);
      DefineT(-1, {}, {{feed1, 0}, {feed2, 0}}, {{add, 3}}, {}, true);
      EXPECT_EQ(TF_OUT_OF_RANGE, TF_GetCode(s_));
      EXPECT_EQ(string("Node 'add' (type: 'AddN', num of outputs: 1) does "
                       "not have output 3\n\tEncountered while processing "
                       "output 0 from function 'MyFunc'"),
                string(TF_Message(s_)));
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jul 20 22:08:54 UTC 2023
    - 63.6K bytes
    - Viewed (0)
  9. tensorflow/compiler/jit/encapsulate_subgraphs_pass_test.cc

    }
    
    Node* AddNLike(const std::vector<ops::NodeOut>& inputs,
                   const GraphDefBuilder::Options& opts) {
      if (opts.HaveError()) return nullptr;
      NodeBuilder node_builder(opts.GetNameForOp("AddN"), "AddNLikeTest",
                               opts.op_registry());
      node_builder.Input(inputs);
      return opts.FinalizeBuilder(&node_builder);
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 113.3K bytes
    - Viewed (0)
  10. tensorflow/compiler/jit/mark_for_compilation_pass.cc

    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 12:19:41 UTC 2024
    - 85.3K bytes
    - Viewed (0)
Back to top