Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/ops/
Dspecial_math_ops_test.py268 _ = special_math_ops.einsum(axes, *inputs)
277 _ = special_math_ops.einsum(
289 _ = special_math_ops.einsum('ii->', x)
299 _ = special_math_ops.einsum(axes, *inputs)
312 output_tensor = special_math_ops.einsum(axes, *input_tensors)
322 correct_value = np.einsum(axes, *input_vals)
330 out = special_math_ops.einsum('ij,jk->ik', m0, m1)
341 out = special_math_ops.einsum('ij,j->i', m0, m1)
353 out = special_math_ops.einsum('ijk,kl->ijl', m0, m1)
364 out = special_math_ops.einsum('kl,ijk->ijl', m0, m1)
[all …]
Dspecial_math_ops.py135 def einsum(equation, *inputs, **kwargs): function
/external/tensorflow/tensorflow/python/ops/linalg/
Dlinalg_impl.py44 einsum = special_math_ops.einsum variable
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.linalg.pbtxt108 name: "einsum"
Dtensorflow.pbtxt568 name: "einsum"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.linalg.pbtxt108 name: "einsum"
Dtensorflow.pbtxt1148 name: "einsum"
/external/tensorflow/tensorflow/compiler/tf2xla/python/
Dxla.py307 einsum = gen_xla_ops.xla_einsum variable
/external/tensorflow/tensorflow/python/keras/layers/
Drecurrent_test.py1373 h = special_math_ops.einsum('bij,ijkl->bkl', inputs, self.kernel)
1375 output = h + special_math_ops.einsum('bij,ijkl->bkl', prev_output,
1435 output_2_3 = special_math_ops.einsum('bij,ijkl->bkl', flatten_inputs[1],
/external/tensorflow/
DRELEASE.md37 * `tf.einsum()`raises `ValueError` for unsupported equations like `"ii->"`.