- Sort Score
- Result 10 results
- Languages All
Results 1 - 8 of 8 for n_init_values (0.23 sec)
-
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) -
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) -
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) -
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) -
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) -
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) -
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) -
tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K bytes - Viewed (0)