- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for matmul (0.07 sec)
-
tensorflow/c/eager/c_api_remote_test_util.cc
TFE_OpAddInput(matmul, h0_task0, status); ASSERT_EQ(TF_GetCode(status), TF_OK) << TF_Message(status); TFE_OpAddInput(matmul, has_packed_input ? packed_handle : h1_task2, status); ASSERT_EQ(TF_GetCode(status), TF_OK) << TF_Message(status); } else { // Handles are on task0 (local), and task2, but op is on task1. matmul = MatMulOp(ctx, h0_task0, h1_task2); } if (remote) {
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Fri Dec 11 22:56:03 UTC 2020 - 9.1K bytes - Viewed (0) -
tensorflow/c/eager/c_api_remote_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);
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Wed Aug 12 00:14:22 UTC 2020 - 5.4K bytes - Viewed (0) -
tensorflow/c/eager/c_api_remote_test_util.h
==============================================================================*/ #ifndef TENSORFLOW_C_EAGER_C_API_REMOTE_TEST_UTIL_H_ #define TENSORFLOW_C_EAGER_C_API_REMOTE_TEST_UTIL_H_ // Run a function containing a MatMul op and check its output. // If heavy_load_on_streaming_rpc is true, send some rpc requests before the one // which creates a remote input, to simulate a scenario that the remote input
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Fri Dec 11 22:56:03 UTC 2020 - 1.4K bytes - Viewed (0) -
tensorflow/c/eager/c_api_test_util.h
// Return an add op multiplying `a` by `b`. TFE_Op* AddOp(TFE_Context* ctx, TFE_TensorHandle* a, TFE_TensorHandle* b); // Return a matmul op multiplying `a` by `b`. TFE_Op* MatMulOp(TFE_Context* ctx, TFE_TensorHandle* a, TFE_TensorHandle* b); // Return an identity op. TFE_Op* IdentityOp(TFE_Context* ctx, TFE_TensorHandle* a); // Return a shape op fetching the shape of `a`.
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Mon Jul 17 23:43:59 UTC 2023 - 7.7K bytes - Viewed (0) -
tensorflow/c/eager/gradient_checker_test.cc
absl::Span<AbstractTensorHandle* const> inputs, absl::Span<AbstractTensorHandle*> outputs) { return ops::MatMul(ctx, inputs[0], inputs[1], &outputs[0], /*transpose_a=*/false, /*transpose_b=*/false, "MatMul"); } absl::Status MulModel(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> inputs,
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 6.5K bytes - Viewed (0) -
src/cmd/asm/internal/arch/mips.go
} // IsMIPSMUL reports whether the op (as defined by an mips.A* constant) is // one of the MUL/DIV/REM/MADD/MSUB instructions that require special handling. func IsMIPSMUL(op obj.As) bool { switch op { case mips.AMUL, mips.AMULU, mips.AMULV, mips.AMULVU, mips.ADIV, mips.ADIVU, mips.ADIVV, mips.ADIVVU, mips.AREM, mips.AREMU, mips.AREMV, mips.AREMVU, mips.AMADD, mips.AMSUB: return true } return false }
Registered: Tue Nov 05 11:13:11 UTC 2024 - Last Modified: Wed Mar 04 19:06:44 UTC 2020 - 1.7K bytes - Viewed (0)