Searched refs:tensordot (Results 1 - 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| H A D | tensordot_op_test.py | 49 math_ops.tensordot(a, b, (a_axes, b_axes))
57 output = math_ops.tensordot(a_ph, b_ph, axes_ph)
71 math_ops.tensordot(a, b, axes_value)
74 math_ops.tensordot(a, b, [[0], [7]])
80 output = math_ops.tensordot(a_ph, b_ph, axes_ph)
98 np_ans = np.tensordot(np_a, np_b, axes_value)
102 tf_ans = math_ops.tensordot(tf_a, tf_b, axes_value).eval()
111 output = math_ops.tensordot(a, b, axes)
115 output = math_ops.tensordot(a, b, axes)
125 output = math_ops.tensordot(
[all...] |
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| H A D | linalg_impl.py | 52 tensordot = math_ops.tensordot variable
|
| /external/tensorflow/tensorflow/python/layers/ |
| H A D | core.py | 155 outputs = standard_ops.tensordot(inputs, self.kernel, [[len(shape) - 1],
|
| /external/tensorflow/tensorflow/python/ops/ |
| H A D | image_ops_impl.py | 1274 gray_float = math_ops.tensordot(flt_image, rgb_weights, [-1, -1])
1813 return math_ops.tensordot(images, kernel, axes=[[ndims - 1], [0]])
1839 return math_ops.tensordot(images, kernel, axes=[[ndims - 1], [0]])
1865 return math_ops.tensordot(images, kernel, axes=[[ndims - 1], [0]])
1891 return math_ops.tensordot(images, kernel, axes=[[ndims - 1], [0]])
|
| H A D | math_ops.py | 101 @@tensordot
2795 @tf_export("tensordot", "linalg.tensordot")
2796 def tensordot(a, b, axes, name=None): function
2807 This operation corresponds to `numpy.tensordot(a, b, axes)`.
2848 This method is helpful in reducing `math_ops.tensordot` to `math_ops.matmul`
|
| /external/tensorflow/tensorflow/contrib/model_pruning/python/layers/ |
| H A D | core_layers.py | 458 outputs = standard_ops.tensordot(inputs, self.masked_kernel,
|
| /external/tensorflow/tensorflow/contrib/factorization/python/ops/ |
| H A D | kmeans_test.py | 327 self.true_score = len(self.points) - np.tensordot(
395 true_score = len(points) - np.tensordot(
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| H A D | kmeans_test.py | 335 self.true_score = len(self.points) - np.tensordot(
402 true_score = len(points) - np.tensordot(
|
| /external/tensorflow/tensorflow/contrib/bayesflow/python/ops/ |
| H A D | layers_dense_variational.py | 244 # To handle broadcasting, we must use `tensordot`.
245 return standard_ops.tensordot(inputs, kernel, axes=[[-1], [0]])
|
Completed in 495 milliseconds