Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for matmul (0.07 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. 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)
  6. 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)
Back to top