- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 45 for x1_shape (0.22 sec)
-
tensorflow/cc/gradients/math_grad_test.cc
auto y = Sub(scope_, x1, x2); RunTest({x1, x2}, {x1_shape, x2_shape}, {y}, {x1_shape}); } TEST_F(NaryGradTest, Mul) { TensorShape x1_shape({3, 2, 5}); TensorShape x2_shape({2, 5}); auto x1 = Placeholder(scope_, DT_FLOAT, Placeholder::Shape(x1_shape)); auto x2 = Placeholder(scope_, DT_FLOAT, Placeholder::Shape(x2_shape)); auto y = Mul(scope_, x1, x2);
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/gradients/data_flow_grad_test.cc
TEST_F(DataFlowGradTest, DynamicStitchGrad) { TensorShape d1_shape({2}); TensorShape d2_shape({2, 2}); std::vector<Output> indices = {Const(scope_, 2), Const(scope_, {1, 0})}; std::vector<Output> data = { Placeholder(scope_, DT_FLOAT, Placeholder::Shape(d1_shape)), Placeholder(scope_, DT_FLOAT, Placeholder::Shape(d2_shape))}; auto y = DynamicStitch(scope_, indices, data); TensorShape y_shape({3, 2});
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 02 18:49:13 UTC 2018 - 2.6K bytes - Viewed (0) -
tensorflow/cc/gradients/linalg_grad_test.cc
auto y = Einsum(scope_, {x}, "ij->ji"); TensorShape y_shape({3, 2}); RunTest({x}, {x_shape}, {y}, {y_shape}); } TEST_F(LinalgGradTest, Einsum_TransposeBroadcast) { TensorShape x_shape({3, 2, 3}); Output x = Placeholder(scope_, DT_FLOAT, Placeholder::Shape(x_shape)); auto y = Einsum(scope_, {x}, "...ij->...ji"); TensorShape y_shape({3, 3, 2}); RunTest({x}, {x_shape}, {y}, {y_shape}); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 07 23:11:54 UTC 2022 - 5.8K bytes - Viewed (0) -
tensorflow/cc/framework/gradient_checker_test.cc
xs.push_back(Placeholder(scope, DT_DOUBLE, Placeholder::Shape(x_shape))); xs.push_back(Placeholder(scope, DT_DOUBLE, Placeholder::Shape(x_shape))); auto y = Stack(scope, xs, Stack::Axis(0)); TensorShape y_shape({2, 1, 2, 3}); double max_error; TF_ASSERT_OK((ComputeGradientError<double, double, double>( scope, xs, {x_shape, x_shape}, {y}, {y_shape}, &max_error))); EXPECT_LT(max_error, 1e-10); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Aug 06 15:54:08 UTC 2018 - 6.7K bytes - Viewed (0) -
tensorflow/cc/gradients/array_grad_test.cc
RunTest(x, x_shape, y, y_shape); } TEST_F(ArrayGradTest, DiagPartGrad) { TensorShape x_shape({5, 2, 5, 2}); auto x = Placeholder(scope_, DT_FLOAT, Placeholder::Shape(x_shape)); auto y = DiagPart(scope_, x); TensorShape y_shape({5, 2}); RunTest(x, x_shape, y, y_shape); } TEST_F(ArrayGradTest, MatrixDiagGrad) { TensorShape x_shape({5, 2});
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 10 23:33:32 UTC 2023 - 19.3K bytes - Viewed (0) -
tensorflow/cc/gradients/nn_grad_test.cc
auto y = AvgPool3D(scope_, x, ksize, strides, "SAME"); RunTest(x, x_shape, y, y_shape); } TEST_F(NNGradTest, LRN) { TensorShape x_shape({1, 1, 2, 1}); auto x = Placeholder(scope_, DT_FLOAT, Placeholder::Shape(x_shape)); auto y = LRN(scope_, x); RunTest(x, x_shape, y, x_shape); } TEST_F(NNGradTest, SoftplusGrad) { TensorShape shape({3, 7});
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/functional_grad_test.cc
FunctionGradTest() : scope_(Scope::NewRootScope()) {} void RunTest(const Output& x, const TensorShape& x_shape, const Output& y, const TensorShape& y_shape) { TF_ASSERT_OK(scope_.status()); float max_error; auto result = (ComputeGradientError<float, float, float>( scope_, {x}, {x_shape}, {y}, {y_shape}, &max_error)); TF_CHECK_OK(result); TF_ASSERT_OK(result); EXPECT_LT(max_error, 1e-3); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 26 14:42:24 UTC 2021 - 3K bytes - Viewed (0) -
tensorflow/c/tf_shape.h
#endif // An opaque type corresponding to a shape in tensorflow. In the future, // we may expose the ABI of TF_Shape for performance reasons. typedef struct TF_Shape TF_Shape; // Return a new, unknown rank shape object. The caller is responsible for // calling TF_DeleteShape to deallocate and destroy the returned shape. TF_CAPI_EXPORT extern TF_Shape* TF_NewShape(); // Returns the rank of `shape`. If `shape` has unknown rank, returns -1.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 30 00:25:41 UTC 2020 - 1.7K bytes - Viewed (0) -
tensorflow/c/tf_shape.cc
#include "tensorflow/c/tf_shape.h" #include <stdint.h> #include "tensorflow/c/tf_shape_internal.h" #include "tensorflow/core/framework/tensor_shape.h" extern "C" { TF_Shape* TF_NewShape() { return tensorflow::wrap(new tensorflow::PartialTensorShape()); } int TF_ShapeDims(const TF_Shape* shape) { return tensorflow::unwrap(shape)->dims(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 30 00:25:41 UTC 2020 - 1.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc
// https://www.tensorflow.org/api_docs/cc/class/tensorflow/ops/batch-mat-mul int64_t x_row_dim = x_shape[x_shape.size() - 2]; int64_t x_col_dim = x_shape[x_shape.size() - 1]; int64_t y_row_dim = y_shape[y_shape.size() - 2]; int64_t y_col_dim = y_shape[y_shape.size() - 1]; int64_t out_row_dim = output_shape[output_shape.size() - 2]; int64_t out_col_dim = output_shape[output_shape.size() - 1];
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 146.7K bytes - Viewed (0)