- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 16 for BatchMatMulV2 (0.3 sec)
-
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf-BatchMatMulV2.mlir
// RUN: tf-opt -xla-legalize-tf %s | FileCheck %s //===----------------------------------------------------------------------===// // tf.BatchMatMulV2 op legalizations. //===----------------------------------------------------------------------===// func.func @batchmatmulv2_basic(%arg0: tensor<1x4x2xf32>, %arg1: tensor<3x2x4xf32>) -> tensor<3x4x4xf32> { // CHECK-LABEL: func @batchmatmulv2_basic
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 06 15:32:52 UTC 2024 - 5.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/prepare-composite-functions-tf.mlir
%0 = "tf.BatchMatMulV2"(%arg0, %arg3) {adj_x = false, adj_y = false} : (tensor<?x8x8xf32>, tensor<8x40xf32>) -> tensor<?x8x40xf32> %1 = "tf.Add"(%0, %arg5) : (tensor<?x8x40xf32>, tensor<40xf32>) -> tensor<?x8x40xf32> %2 = "tf.BatchMatMulV2"(%1, %arg4) {adj_x = false, adj_y = true} : (tensor<?x8x40xf32>, tensor<10x40xf32>) -> tensor<?x8x10xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 122.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/end2end/unroll_batch_matmul_disabled.pbtxt
key: "shape" value { shape { dim { size: 3 } dim { size: 7 } } } } } node { name: "MatMul" op: "BatchMatMulV2" input: "Placeholder" input: "Placeholder_1" attr { key: "T" value { type: DT_FLOAT } } attr { key: "adj_x" value { b: false } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 1.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/end2end/unroll_batch_matmul.pbtxt
key: "shape" value { shape { dim { size: 3 } dim { size: 7 } } } } } node { name: "MatMul" op: "BatchMatMulV2" input: "Placeholder" input: "Placeholder_1" attr { key: "T" value { type: DT_FLOAT } } attr { key: "adj_x" value { b: false } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 2.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf-with-tf2xla-hlo-importer.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 06 15:32:52 UTC 2024 - 38.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/legalize_tf_test.cc
} INSTANTIATE_TEST_SUITE_P( BatchMatMulTest, BatchMatMulTest, ::testing::ValuesIn<MatMulTestCase>({ {"BatchMatMul"}, {"BatchMatMulV2"}, {"BatchMatMulV3"}, }), [](const ::testing::TestParamInfo<BatchMatMulTest::ParamType>& info) { return info.param.mat_mul_method; }); TEST(LegalizeTFTest, DumpsProducedHLO) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 13 23:59:33 UTC 2024 - 16.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/einsum.cc
for (int64_t i = 0; i < input.size(); ++i) { output[permutation[i]] = input[i]; } return output; } // Computes the transpositions required to convert dnums to one supported by // tf.BatchMatmulV2 and returns the new set of dimension numbers with them. // Transposed LHS shape will be B0,...,Bn,L0,...,Ln,C0,...,Cn and, // transposed RHS shape will be B0,...,Bn,C0,...,Cn,R0,...,Rn respectively.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 33.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_tf.cc
// operands are properly supported in declarative rewrite rule specification. DECL_CONVERT_OP(Assert); DECL_CONVERT_OP(ConcatV2); DECL_CONVERT_OP(BatchMatMul); DECL_CONVERT_OP(BatchMatMulV2); DECL_CONVERT_OP(BatchMatMulV3); DECL_CONVERT_OP(MatMul); DECL_CONVERT_OP(MatrixDiagV2); DECL_CONVERT_OP(MatrixDiagV3); DECL_CONVERT_OP(Pack); DECL_CONVERT_OP(Split); DECL_CONVERT_OP(SplitV);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 20:06:54 UTC 2024 - 45.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/legalize-tf.mlir
} func.func @matmul_batchv2(%arg0: tensor<2x10x15xf32>, %arg1: tensor<15x17xf32>) -> tensor<2x10x17xf32> { %0 = "tf.BatchMatMulV2"(%arg0, %arg1) {T = "tfdtype$DT_FLOAT", device = "/device:CPU:0", name = "MatMul", adj_x = false, adj_y = false} : (tensor<2x10x15xf32>, tensor<15x17xf32>) -> tensor<2x10x17xf32> func.return %0 : tensor<2x10x17xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 05 01:54:33 UTC 2024 - 153.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/canonicalize.mlir
func.return %0: tensor<2x3xf32> } // CHECK-LABEL: testBatchMatMulToV2 func.func @testBatchMatMulToV2(%arg0: tensor<2x3x5xf32>, %arg1: tensor<2x5x7xf32>) -> tensor<2x3x7xf32> { // CHECK: "tf.BatchMatMulV2"(%arg0, %arg1) <{adj_x = false, adj_y = false, grad_x = false, grad_y = false}> {device = "/job:localhost/replica:0/task:0/device:GPU:0"}
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 132.1K bytes - Viewed (0)