Home
last modified time | relevance | path

Searched refs:einsum (Results 1 – 25 of 33) sorted by relevance

12

/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_Einsum.pbtxt18 String describing the Einstein Summation operation; in the format of np.einsum.
35 call `tf.einsum` directly. It is a hidden Op used by `tf.einsum`.
59 (d) Contraction: In case of binary einsum, axes corresponding to labels
70 Note: This operation is not supported by `np.einsum` or `tf.einsum`; it is
71 provided to enable computing the symbolic gradient of `tf.einsum`.
89 Similar to [`numpy.einsum`](https://docs.scipy.org/doc/numpy/reference/generated/numpy.einsum.html).
91 Comparison with `numpy.einsum`:
93 * This Op only supports unary and binary forms of `numpy.einsum`.
96 supported by `numpy.einsum`.
/external/tensorflow/tensorflow/python/kernel_tests/
Deinsum_op_test.py53 a = np.einsum(s, *inputs)
54 b = self.evaluate(gen_linalg_ops.einsum(input_tensors, s))
174 a = np.einsum(equation,
177 a = np.einsum(equation, *inputs)
179 b = self.evaluate(gen_linalg_ops.einsum(input_tensors, equation))
208 _ = self.evaluate(gen_linalg_ops.einsum(args[1:], args[0]))
214 _ = self.evaluate(gen_linalg_ops.einsum(placeholders, args[0]))
231 a = np.einsum(equation, *inputs)
232 b = self.evaluate(gen_linalg_ops.einsum(input_placeholders, equation))
250 b = self.evaluate(gen_linalg_ops.einsum([np.diag(a)], s))
[all …]
/external/tensorflow/tensorflow/compiler/tests/
Deinsum_op_test.py58 lambda x, y: special_math_ops.einsum('ij,jk->ik', x, y),
66 lambda x, y: special_math_ops.einsum('ijk,kji', x, y),
74 lambda x, y: special_math_ops.einsum('ij,j->', x, y),
82 lambda x: special_math_ops.einsum('ijk->kji', x),
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dxla_control_flow_ops_test.py57 z1 = xla_ops.einsum(x_i, y, "ab,bc->ac")
58 z2 = xla_ops.einsum(x, y_i, "ab,bc->ac")
59 z3 = xla_ops.einsum(x, y, "ab,bc->ac")
60 z4 = xla_ops.einsum(x_i, y_i, "ab,bc->ac")
61 z5 = xla_ops.einsum(y_i, x_i, "cd,ce->de") # Includes transpose.
Dmath_test.py772 z1 = special_math_ops.einsum("ab,bc->ac", x_i, y)
773 z2 = special_math_ops.einsum("ab,bc->ac", x, y_i)
774 z3 = special_math_ops.einsum("ab,bc->ac", x, y)
775 z4 = special_math_ops.einsum("ab,bc->ac", x_i, y_i)
776 z5 = special_math_ops.einsum("cd,ce->de", y_i, x_i) # Includes transpose.
/external/tensorflow/tensorflow/python/ops/
Dspecial_math_ops_test.py652 a = np.einsum(s, *inputs)
653 b = self.evaluate(special_math_ops.einsum(s, *input_tensors))
661 _ = special_math_ops.einsum(
798 a = np.einsum(equation, *inputs)
799 b = self.evaluate(special_math_ops.einsum(equation, *input_placeholders))
819 a = np.einsum(s, x, y, z)
820 b = self.evaluate(special_math_ops.einsum(s, x, y, z))
925 _ = special_math_ops.einsum(*args)
931 _ = self.evaluate(special_math_ops.einsum(args[0], *placeholders))
943 output = self.evaluate(special_math_ops.einsum(equation, *input_tensors))
[all …]
Dspecial_math_ops.py608 def einsum(equation, *inputs, **kwargs): function
1180 return gen_linalg_ops.einsum(inputs, resolved_equation)
1199 inputs.append(gen_linalg_ops.einsum(operands, binary_equation))
Dlinalg_grad.py216 return gen_linalg_ops.einsum([broadcasted_grad],
282 grad_reduced = gen_linalg_ops.einsum([output_grad, other_operand],
314 return gen_linalg_ops.einsum([grad],
/external/tensorflow/third_party/
Dopt_einsum.BUILD2 # opt_einsum is a library for optimizing tensor contraction order for einsum.
/external/tensorflow/tensorflow/python/keras/layers/
Dmulti_head_attention.py461 attention_scores = special_math_ops.einsum(self._dot_product_equation, key,
472 attention_output = special_math_ops.einsum(self._combine_equation,
Deinsum_dense.py201 ret = special_math_ops.einsum(self.equation, inputs, self.kernel)
Drecurrent_test.py1820 h = special_math_ops.einsum('bij,ijkl->bkl', inputs, self.kernel)
1822 output = h + special_math_ops.einsum('bij,ijkl->bkl', prev_output,
1882 output_2_3 = special_math_ops.einsum('bij,ijkl->bkl', flatten_inputs[1],
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dbatchmatmul_to_einsum.mlir1 // RUN: tf-opt %s -tf-batch-matmul-to-tf-einsum | FileCheck %s
Deinsum.mlir1 // RUN: tf-opt -split-input-file -verify-diagnostics -tf-einsum %s | FileCheck %s
/external/tensorflow/tensorflow/python/tpu/
Dtpu_test.py175 return special_math_ops.einsum("abc,acd->abd", a, b)
/external/tensorflow/tensorflow/lite/g3doc/convert/
Doperation_fusion.md21 [tf.einsum](https://www.tensorflow.org/api_docs/python/tf/einsum). A primitive
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.linalg.pbtxt132 name: "einsum"
/external/tensorflow/tensorflow/python/ops/linalg/
Dlinalg_impl.py49 einsum = special_math_ops.einsum variable
/external/tensorflow/tensorflow/compiler/tf2xla/python/
Dxla.py322 einsum = gen_xla_ops.xla_einsum variable
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/
DBUILD623 "transforms/einsum.h",
880 "transforms/einsum.cc",
954 "transforms/einsum.h",
/external/tensorflow/tensorflow/python/ops/numpy_ops/
Dnp_math_ops.py1334 def einsum(subscripts, *operands, **kwargs): # pylint: disable=missing-docstring function
1356 res = special_math_ops.einsum(subscripts, *operands, optimize=tf_optimize)
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
Dexport.mlir462 // Simple einsum is lowered to HLO dot op.
464 …%0 = "mhlo.einsum"(%arg0, %arg1) {einsum_config = "ab,bc->ac"} : (tensor<3x4xi32>, tensor<4x5xi32>…
/external/tensorflow/third_party/py/numpy/tf_numpy_api/
Dtensorflow.experimental.numpy.pbtxt364 name: "einsum"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.pbtxt624 name: "einsum"
/external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/
Dhlo_ops.td976 def HLO_EinsumOp: HLO_Op<"einsum", [NoSideEffect]>, BASE_EinsumOp {

12