| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | tensordot_op_test.py | 52 math_ops.tensordot(a, b, (a_axes, b_axes))
60 output = math_ops.tensordot(a_ph, b_ph, axes_ph)
75 math_ops.tensordot(a, b, axes_value)
78 math_ops.tensordot(a, b, [[0], [7]])
84 output = math_ops.tensordot(a_ph, b_ph, axes_ph)
102 np_ans = np.tensordot(np_a, np_b, axes_value)
106 tf_ans = math_ops.tensordot(tf_a, tf_b, axes_value)
116 output = math_ops.tensordot(a, b, axes)
120 output = math_ops.tensordot(a, b, axes)
130 output = math_ops.tensordot(a, b, axes)
[all …]
|
| /external/tensorflow/tensorflow/contrib/opt/python/training/ |
| D | shampoo_test.py | 193 np.tensordot(grad_np, grad_np, axes=([1, 2], [1, 2])) /
197 np.tensordot(grad_np, grad_np, axes=([0, 2], [0, 2])) /
201 np.tensordot(grad_np, grad_np, axes=([0, 1], [0, 1])) /
205 precond_grad = np.tensordot(grad_np, mat_g1_a, axes=([0], [0]))
206 precond_grad = np.tensordot(precond_grad, mat_g2_a, axes=([0], [0]))
207 precond_grad = np.tensordot(precond_grad, mat_g3_a, axes=([0], [0]))
218 np.tensordot(grad_np_2, grad_np_2, axes=([1, 2], [1, 2])) /
222 np.tensordot(grad_np_2, grad_np_2, axes=([0, 2], [0, 2])) /
226 np.tensordot(grad_np_2, grad_np_2, axes=([0, 1], [0, 1])) /
230 precond_grad = np.tensordot(grad_np_2, mat_g1_a, axes=([0], [0]))
[all …]
|
| D | shampoo.py | 226 grad_outer = math_ops.tensordot(grad, grad, axes=(axes, axes),
384 preconditioned_grad = math_ops.tensordot(preconditioned_grad, mat_h,
|
| /external/tensorflow/tensorflow/contrib/constrained_optimization/python/ |
| D | swap_regret_optimizer_test.py | 110 matrix_eigenvector1 = np.tensordot(matrix1, eigenvector1, axes=1)
111 matrix_eigenvector2 = np.tensordot(matrix2, eigenvector2, axes=1)
|
| D | external_regret_optimizer.py | 257 objective + standard_ops.tensordot(
|
| D | swap_regret_optimizer.py | 369 loss = standard_ops.tensordot(
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linalg_impl.py | 57 tensordot = math_ops.tensordot variable
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| D | kmeans_test.py | 334 self.true_score = len(self.points) - np.tensordot(
401 true_score = len(points) - np.tensordot(
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | image_ops_impl.py | 1748 gray_float = math_ops.tensordot(flt_image, rgb_weights, [-1, -1])
2622 return math_ops.tensordot(images, kernel, axes=[[ndims - 1], [0]])
2649 return math_ops.tensordot(images, kernel, axes=[[ndims - 1], [0]])
2676 return math_ops.tensordot(images, kernel, axes=[[ndims - 1], [0]])
2703 return math_ops.tensordot(images, kernel, axes=[[ndims - 1], [0]])
|
| D | math_ops.py | 3553 def tensordot(a, b, axes, name=None): function
|
| /external/tensorflow/tensorflow/contrib/factorization/python/ops/ |
| D | kmeans_test.py | 366 self.true_score = len(self.points) - np.tensordot(
434 true_score = len(points) - np.tensordot(
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.linalg.pbtxt | 192 name: "tensordot"
|
| D | tensorflow.pbtxt | 1016 name: "tensordot"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.linalg.pbtxt | 192 name: "tensordot"
|
| D | tensorflow.pbtxt | 2284 name: "tensordot"
|
| /external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| D | xla.py | 279 return math_ops.tensordot(lhs, rhs, axes=1, name=name)
|
| /external/tensorflow/tensorflow/contrib/model_pruning/python/layers/ |
| D | core_layers.py | 459 outputs = standard_ops.tensordot(inputs, self.masked_kernel,
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | core.py | 1017 outputs = standard_ops.tensordot(inputs, self.kernel, [[rank - 1], [0]])
|
| /external/tensorflow/tensorflow/contrib/tpu/python/tpu/ |
| D | keras_support.py | 288 return math_ops.tensordot(inputs, self.embeddings, 1)
|