Searched defs:einsum (Results 1 – 7 of 7) sorted by relevance
619 def einsum(equation, *inputs, **kwargs): function
395 mlir::TF::EinsumOp einsum = mlir::cast<mlir::TF::EinsumOp>(op); in MaybeRelayoutInputs() local
46 einsum = special_math_ops.einsum variable
368 einsum = gen_xla_ops.xla_einsum variable
1349 def einsum(subscripts, *operands, **kwargs): # pylint: disable=missing-docstring function
878 string einsum; member
3392 auto einsum = ops::Einsum(s.WithOpName("my_einsum"), inputs, eq); in TEST_P() local