- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for BatchMatMulV2 (0.28 sec)
-
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/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/lite/ir/tfl_ops.td
let summary = "Batch Matrix Multiply Operator"; let description = [{ Performs a batched matrix multiplication on the inputs. Follows the conventions of TensorFlow BatchMatMulV2, with support for unknown dimensions in the batch dimensions and broadcasting. Inputs: `inputs[0]`: required: input LHS `inputs[1]`: required: input RHS
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 19:09:08 UTC 2024 - 186K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
Value *out_lhs, Value *out_rhs, PatternRewriter *rewriter) { // The dimension structure of the relevant operands to a tf.BatchMatMulV2 is: // - lhs: [LHSBATCHDIMS..., LHSROWS, LHSCOLS] // - rhs: [RHSBATCHDIMS..., RHSROWS, RHSCOLS] // - result: [broadcast(LHSBATCHDIMS, RHSBATCHDIMS)..., LHSROWS, RHSCOLS]
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td
); TF_DerivedOperandTypeAttr T = TF_DerivedOperandTypeAttr<0>; let hasCanonicalizer = 1; let hasVerifier = 1; } def TF_BatchMatMulV2Op : TF_Op<"BatchMatMulV2", [Pure, TF_SameOperandsAndResultElementTypeResolveRef]> { let summary = "Multiplies slices of two tensors in batches."; let description = [{ Multiplies all slices of `Tensor` `x` and `y` (each slice can be
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K bytes - Viewed (0) -
RELEASE.md
* Remove unused `StringViewVariantWrapper`. * Delete unused `Fingerprint64Map` op registration * Add broadcasting support to `tf.matmul`. * Add C++ Gradient for `BatchMatMulV2`. * Add `tf.math.cumulative_logsumexp` operation. * Add ellipsis (...) support for `tf.einsum()`. * Add expand_composites argument to all `nest.*` methods. * Added `strings.byte_split`.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 730.3K bytes - Viewed (0)