- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 21 for matmul (0.17 sec)
-
tensorflow/c/experimental/gradients/math_grad.cc
// Calc Grad AbstractTensorHandle* matmul_A_output; AbstractTensorHandle* matmul_B_output; std::string name_grad_A = "MatMul_Grad_A"; std::string name_grad_B = "MatMul_Grad_B"; if (!t_a && !t_b) { TF_RETURN_IF_ERROR(MatMul(ctx, upstream_grad, B.get(), &matmul_A_output, /*transpose_a = */ false,
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 15.2K bytes - Viewed (0) -
tensorflow/c/eager/c_api_test.cc
TFE_TensorHandle* m = TestMatrixTensorHandle(ctx); TFE_Op* matmul = TFE_NewOp(ctx, "MatMul", status); CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); TFE_TensorHandle* retvals[1]; int num_retvals = 1; for (auto s : state) { TFE_OpReset(matmul, "MatMul", nullptr, status); CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); TFE_OpAddInput(matmul, m, status);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Aug 03 20:50:20 GMT 2023 - 94.6K bytes - Viewed (1) -
tensorflow/c/experimental/gradients/math_grad_test.cc
absl::Span<AbstractTensorHandle* const> inputs, absl::Span<AbstractTensorHandle*> outputs) -> Status { return ops::MatMul(ctx, inputs[0], inputs[1], &outputs[0], transpose_a, transpose_b, "MatMul"); }; ASSERT_NO_FATAL_FAILURE(CompareNumericalAndAutodiffGradients( MatMulModel, BuildGradModel(MatMulModel, registry_),
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Thu Apr 13 17:32:14 GMT 2023 - 16.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental_test.cc
// Build an abstract operation. auto* matmul_op = TF_NewAbstractOp(graph_ctx); TF_AbstractOpSetOpType(matmul_op, "MatMul", status.get()); ASSERT_EQ(TF_OK, TF_GetCode(status.get())) << TF_Message(status.get()); TF_AbstractOpSetOpName(matmul_op, "my_matmul", status.get()); ASSERT_EQ(TF_OK, TF_GetCode(status.get())) << TF_Message(status.get());
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri May 19 21:44:52 GMT 2023 - 39.1K bytes - Viewed (0) -
tensorflow/c/c_api_experimental_test.cc
TFE_Context* tfe_context_; }; TEST_F(ShapeInferenceTest, InfersShapesFromInputShapes) { TFE_Op* matmul_op; matmul_op = TFE_NewOp(tfe_context_, "MatMul", status_); CHECK_EQ(TF_OK, TF_GetCode(status_)) << TF_Message(status_); // Infer shape when everything is known. CheckOutputShapes(matmul_op, /*input_shapes*/ {make_shape({3, 2}), make_shape({2, 4})}, /*input_tensors*/ {},
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Jan 17 22:27:52 GMT 2023 - 13.1K bytes - Viewed (1) -
tensorflow/c/eager/c_api_distributed_test.cc
ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); TFE_Op* matmul = MatMulOp(ctx, h0_task1, h1_task1); TFE_OpSetDevice(matmul, remote_device_name, status); EXPECT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); TFE_TensorHandle* retvals[1]; int num_retvals = 1; TFE_Execute(matmul, &retvals[0], &num_retvals, status); EXPECT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 23.5K bytes - Viewed (0) -
tensorflow/c/eager/c_api_cluster_test.cc
TFE_TensorHandle* h0_task0 = TestMatrixTensorHandle(ctx); TFE_Op* matmul = MatMulOp(ctx, h0_task0, h0_task0); TFE_OpSetDevice(matmul, remote_device_name, status); EXPECT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); TFE_TensorHandle* retvals[1]; int num_retvals = 1; TFE_Execute(matmul, &retvals[0], &num_retvals, status); EXPECT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Apr 14 10:03:59 GMT 2023 - 19.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api_experimental_test.cc
EXPECT_EQ(7, product[0]); EXPECT_EQ(10, product[1]); EXPECT_EQ(15, product[2]); EXPECT_EQ(22, product[3]); TF_DeleteStatus(status); } TEST(CAPI, Executor_MatMul_CPU) { Executor_MatMul_CPU(false); } TEST(CAPI, Executor_MatMul_CPUAsync) { Executor_MatMul_CPU(true); } void Deleter(void* data, size_t unused, void* tensor_handle) { TFE_DeleteTensorHandle(static_cast<TFE_TensorHandle*>(tensor_handle)); }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Aug 03 03:14:26 GMT 2023 - 31.5K bytes - Viewed (1) -
tensorflow/c/c_api_test.cc
"gradients/MatMul", false, true); TF_Operation* matmul2 = MatMul(expected_graph_, s_, const0, const3, "gradients/MatMul_1", true, false); expected_grad_outputs[0] = {matmul1, 0}; expected_grad_outputs[1] = {matmul2, 0}; } TF_Tensor* FloatTensor2x2(const float* values) {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 96.9K bytes - Viewed (3) -
tensorflow/c/eager/gradient_checker_test.cc
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Apr 14 10:03:59 GMT 2023 - 6.5K bytes - Viewed (0)