Searched refs:EinsumOp (Results 1 – 18 of 18) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | einsum_op.cc | 57 class EinsumOp : public XlaOpKernel { class
59 explicit EinsumOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in EinsumOp() function in tensorflow::__anon5583fa2c0111::EinsumOp
63 ~EinsumOp() override = default;
93 TF_DISALLOW_COPY_AND_ASSIGN(EinsumOp);
98 REGISTER_XLA_OP(Name("Einsum").TypeConstraint("T", kEinsumTypes), EinsumOp);
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | einsum.h | 43 struct ConvertTFEinsumOp : public OpRewritePattern<TF::EinsumOp> {
46 : OpRewritePattern<TF::EinsumOp>(context) {} in ConvertTFEinsumOp()
48 LogicalResult matchAndRewrite(TF::EinsumOp op,
|
| D | batchmatmul_to_einsum.cc | 76 rewriter.replaceOpWithNewOp<TF::EinsumOp>( in matchAndRewrite()
|
| D | einsum.cc | 306 LogicalResult rewriteToBatchMatmul(TF::EinsumOp op, in rewriteToBatchMatmul()
351 TF::EinsumOp op, PatternRewriter& rewriter) const { in matchAndRewrite()
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | einsum_op_impl_int64.cc | 23 EinsumOp<D##Device, TYPE>);
|
| D | einsum_op_impl_bfloat16.cc | 23 EinsumOp<D##Device, TYPE>);
|
| D | einsum_op_impl_int32.cc | 23 EinsumOp<D##Device, TYPE>);
|
| D | einsum_op_impl_double.cc | 23 EinsumOp<D##Device, TYPE>);
|
| D | einsum_op_impl_float.cc | 23 EinsumOp<D##Device, TYPE>);
|
| D | einsum_op_impl_half.cc | 23 EinsumOp<D##Device, TYPE>);
|
| D | einsum_op_impl_complex64.cc | 23 EinsumOp<D##Device, TYPE>);
|
| D | einsum_op_impl_complex128.cc | 23 EinsumOp<D##Device, TYPE>);
|
| D | einsum_op_impl.h | 590 class EinsumOp : public OpKernel {
592 explicit EinsumOp(OpKernelConstruction* c) : OpKernel(c) { in EinsumOp() function
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | chlo_canonicalize.td | 25 // Convert UnaryEinsumOp to EinsumOp with two operands with redundant first
|
| D | mhlo_canonicalize.td | 25 // Convert UnaryEinsumOp to EinsumOp with two operands with redundant first
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | hlo_ops.td | 999 // UnaryEinsumOp is unconditionally canonicalized to the binary EinsumOp so
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 1689 class ConvertEinsumOp : public OpRewritePattern<TF::EinsumOp> {
1693 LogicalResult matchAndRewrite(TF::EinsumOp op, in matchAndRewrite()
1701 rewriter.replaceOpWithNewOp<EinsumOp>(op, op.getType(), inputs[0], in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_a_m.cc | 2042 static LogicalResult Verify(EinsumOp op) { in Verify()
|