- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for in_shape (0.82 sec)
-
tensorflow/cc/gradients/linalg_grad_test.cc
TensorShape y_shape({3, 3, 2}); RunTest({x}, {x_shape}, {y}, {y_shape}); } TEST_F(LinalgGradTest, Einsum_MatMul) { TensorShape x_shape({2, 3}); TensorShape y_shape({3, 3}); Output x = Placeholder(scope_, DT_FLOAT, Placeholder::Shape(x_shape)); Output y = Placeholder(scope_, DT_FLOAT, Placeholder::Shape(y_shape)); auto z = Einsum(scope_, {x, y}, "ij,jk->ik"); TensorShape z_shape({2, 3});
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/gradients/image_grad.cc
// The internal gradient implementation needs the shape of the input image. // x_shape = shape(x)[1:3] // = slice(shape(x), {1}, {3 - 1}) auto x_shape = Slice(scope, Shape(scope, op.input(0)), {1}, {2}); grad_outputs->push_back(internal::ResizeNearestNeighborGrad( scope, grad_inputs[0], x_shape, internal::ResizeNearestNeighborGrad::AlignCorners(align_corners)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 11 00:29:23 UTC 2021 - 5.7K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental.h
// Represents a (partially-defined) shape. typedef struct TF_Shape { int num_dims; // Must be >= -1; -1 represents unknown rank. int64_t* dim_sizes; } TF_Shape; // Add a new parameter to a TensorFlow Function. TF_AbstractTensor* TF_AddFunctionParameter(TF_ExecutionContext* func, TF_DataType dtype, TF_Shape shape, TF_Status* s);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Oct 24 11:27:00 UTC 2021 - 7K 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/c/eager/c_api_unified_experimental.cc
TF_DeleteExecutionContext(ctx); return wrap(func); } TF_AbstractTensor* TF_AddFunctionParameter(TF_ExecutionContext* func, TF_DataType dtype, TF_Shape shape, TF_Status* s) { DCHECK_GE(shape.num_dims, -1); TracingTensorHandle* t; TracingContext* tracing_ctx = dyn_cast<TracingContext>(unwrap(func)); if (!tracing_ctx) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 10:15:17 UTC 2024 - 9K bytes - Viewed (0)