Home
last modified time | relevance | path

Searched refs:tensordot (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/
Dtensordot_op_test.py49 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(a, b, axes)
[all …]
/external/tensorflow/tensorflow/python/ops/linalg/
Dlinalg_impl.py52 tensordot = math_ops.tensordot variable
/external/tensorflow/tensorflow/python/ops/
Dimage_ops_impl.py1274 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]])
Dmath_ops.py2796 def tensordot(a, b, axes, name=None): function
/external/tensorflow/tensorflow/docs_src/api_guides/python/
Dcontrib.signal.md134 mel_spectrograms = tf.tensordot(
136 # Note: Shape inference for `tf.tensordot` does not currently handle this case.
Dmath_ops.md114 * @{tf.tensordot}
/external/tensorflow/tensorflow/tools/api/golden/
Dtensorflow.linalg.pbtxt124 name: "tensordot"
Dtensorflow.pbtxt1984 name: "tensordot"
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Dkmeans_test.py335 self.true_score = len(self.points) - np.tensordot(
402 true_score = len(points) - np.tensordot(
/external/tensorflow/tensorflow/contrib/factorization/python/ops/
Dkmeans_test.py327 self.true_score = len(self.points) - np.tensordot(
395 true_score = len(points) - np.tensordot(
/external/tensorflow/tensorflow/python/layers/
Dcore.py155 outputs = standard_ops.tensordot(inputs, self.kernel, [[len(shape) - 1],
/external/tensorflow/tensorflow/contrib/model_pruning/python/layers/
Dcore_layers.py458 outputs = standard_ops.tensordot(inputs, self.masked_kernel,
/external/tensorflow/tensorflow/contrib/bayesflow/python/ops/
Dlayers_dense_variational.py245 return standard_ops.tensordot(inputs, kernel, axes=[[-1], [0]])