Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 8 of 8 for n_init_values (0.23 sec)

  1. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc

      const auto &init_values_ty = op.getInitValues().getType();
      int n_init_values = init_values_ty.size();
      if (n_init_values != n_inputs) {
        return op.emitOpError() << "Number of inputs (" << n_inputs
                                << ") is different than number of init_values ("
                                << n_init_values << ")";
      }
    
      auto input_ty_0 = inputs_ty[0].cast<ShapedType>();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 22:07:10 UTC 2024
    - 170.8K bytes
    - Viewed (0)
  2. tensorflow/cc/gradients/nn_grad_test.cc

      auto y = Elu(scope_, x);
      Tensor x_init_value = test::AsTensor<float>(
          {-0.9f, -0.7f, -0.5f, -0.3f, -0.1f, 0.1f, 0.3f, 0.5f, 0.7f, 0.9f},
          {5, 2});
      RunTest(x, x_init_value, y, shape);
    }
    
    TEST_F(NNGradTest, SeluGrad) {
      TensorShape shape({5, 2});
      auto x = Placeholder(scope_, DT_FLOAT, Placeholder::Shape(shape));
      auto y = Selu(scope_, x);
      Tensor x_init_value = test::AsTensor<float>(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 22 20:45:22 UTC 2022
    - 15K bytes
    - Viewed (0)
  3. tensorflow/cc/gradients/math_grad_test.cc

        EXPECT_LT(max_error, 1e-3);
      }
    
      void RunTest(const Output& x, const Tensor& x_init_value, const Output& y,
                   const TensorShape& y_shape) {
        TF_ASSERT_OK(scope_.status());
        float max_error;
        TF_ASSERT_OK((ComputeGradientError<float, float, float>(
            scope_, x, x_init_value, y, y_shape, &max_error)));
        EXPECT_LT(max_error, 1e-3);
      }
    
      Scope scope_;
    };
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Aug 25 18:20:20 UTC 2023
    - 36K bytes
    - Viewed (0)
  4. tensorflow/cc/framework/gradient_checker.cc

    Status ComputeGradientError(const Scope& scope, const Output& x,
                                const Tensor& x_init_value, const Output& y,
                                const TensorShape& y_shape, JAC_T* max_error) {
      // Initialize 'x_data' from 'x_init_value'.
      std::vector<Tensor> x_datas(1, Tensor(x_init_value));
      // Compute gradient error.
      return ComputeGradientErrorInternal<X_T, Y_T, JAC_T>(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Apr 13 05:57:22 UTC 2024
    - 18.2K bytes
    - Viewed (0)
  5. tensorflow/cc/framework/gradient_checker.h

    template <typename X_T, typename Y_T, typename JAC_T>
    Status ComputeGradientError(const Scope& scope, const Output& x,
                                const Tensor& x_init_value, const Output& y,
                                const TensorShape& y_shape, JAC_T* max_error);
    
    }  // namespace tensorflow
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Oct 05 15:35:17 UTC 2022
    - 2.8K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tfr/python/tfr_gen.py

        # updated to these new ssa values before it enters the scope of the loop.
        out_types = []
        init_values = []
        for symbol, ssa_value in zip(loop_carried, ret_ssa_values):
          init, ty = self.symbol_table.lookup(symbol)
          self.symbol_table.insert_symbol(symbol, ssa_value, ty)
          out_types.append(str(ty))
          init_values.append((init, ty))
    
        # Create a new scope in case the local variables are leaked.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 27 15:27:03 UTC 2022
    - 55.8K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc

    // set of values. Returns the final set of values via `final_values`. The
    // initial set of values is passed in via `init_values`.
    //
    // This effectively does:
    //
    // ```c++
    // SmallVector<Values, 4> old_values = init_values;
    // SmallVector<Values, 4> new_values;
    // for (int i = 0; i < num_iterations; ++i) {
    //   body_fn(old_values, &new_values, ...);
    //   old_values = new_values;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 291.8K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td

      }];
    
      let arguments = (ins
        Arg<Variadic<TF_Tensor>, [{the input tensor(s)}]>:$inputs,
        Arg<Variadic<TF_Tensor>, [{scalar initial value(s) for the reduction}]>:$init_values,
    
        I64ArrayAttr:$dimensions_to_reduce,
        SymbolRefAttr:$reducer
      );
    
      let results = (outs
        Variadic<TF_Tensor>:$outputs
      );
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 23:24:08 UTC 2024
    - 793K bytes
    - Viewed (0)
Back to top