experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix a misplaced comment.

Browse Source 
PiperOrigin-RevId: 349473783
Change-Id: If4427a6502bc3cde7ddc97ad55879be6142f25f8
This commit is contained in:
Penporn Koanantakool 2020-12-29 14:59:03 -08:00 committed by TensorFlower Gardener
parent 054c7c1e8e
commit 8ea9e8cfab
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/core/kernels/linalg/einsum_op_impl.h
View File

@ -59,12 +59,10 @@ using LabelCounts = gtl::InlinedVector<int, 8>;
using OperandLabelCounts = gtl::InlinedVector<LabelCounts, 2>; using OperandLabelCounts = gtl::InlinedVector<LabelCounts, 2>;
using LabelToDimSizes = gtl::InlinedVector<int64, 8>; using LabelToDimSizes = gtl::InlinedVector<int64, 8>;
// Dummy axis label used to denote an ellipsis in an input or output subscript.
constexpr int kEllipsisLabel = -1; constexpr int kEllipsisLabel = -1;
struct EinsumHelper { struct EinsumHelper {
// Dummy axis label used to denote an ellipsis in an input or output
// subscript.
// Each dimension is categorized into exactly one of five types based on // Each dimension is categorized into exactly one of five types based on
// whether its corresponding label is present in the input and/or the output // whether its corresponding label is present in the input and/or the output
// subscripts. // subscripts.
@ -538,7 +536,7 @@ struct EinsumHelper {
return CopyFrom(input, output_shape, output); return CopyFrom(input, output_shape, output);
} }
// Contracts the inputs along the last axis. (or the second last if the // Contracts the inputs along the last axis (or the second last if the
// corresponding value of swap_free_and_contract is true). The batch // corresponding value of swap_free_and_contract is true). The batch
// dimensions are broadcast to the output shape. // dimensions are broadcast to the output shape.
// TODO(anudhyan): BatchMatMul might devolve into a component-wise // TODO(anudhyan): BatchMatMul might devolve into a component-wise