- Sort Score
- Result 10 results
- Languages All
Results 31 - 34 of 34 for BatchMatMulV2 (0.25 sec)
-
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)