Searched refs:tensordot (Results 1 – 13 of 13) sorted by relevance
/external/tensorflow/tensorflow/python/kernel_tests/ |
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(a, b, axes) [all …]
|
/external/tensorflow/tensorflow/python/ops/linalg/ |
D | linalg_impl.py | 52 tensordot = math_ops.tensordot variable
|
/external/tensorflow/tensorflow/python/ops/ |
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]])
|
D | math_ops.py | 2796 def tensordot(a, b, axes, name=None): function
|
/external/tensorflow/tensorflow/docs_src/api_guides/python/ |
D | contrib.signal.md | 134 mel_spectrograms = tf.tensordot( 136 # Note: Shape inference for `tf.tensordot` does not currently handle this case.
|
D | math_ops.md | 114 * @{tf.tensordot}
|
/external/tensorflow/tensorflow/tools/api/golden/ |
D | tensorflow.linalg.pbtxt | 124 name: "tensordot"
|
D | tensorflow.pbtxt | 1984 name: "tensordot"
|
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
D | kmeans_test.py | 335 self.true_score = len(self.points) - np.tensordot( 402 true_score = len(points) - np.tensordot(
|
/external/tensorflow/tensorflow/contrib/factorization/python/ops/ |
D | kmeans_test.py | 327 self.true_score = len(self.points) - np.tensordot( 395 true_score = len(points) - np.tensordot(
|
/external/tensorflow/tensorflow/python/layers/ |
D | core.py | 155 outputs = standard_ops.tensordot(inputs, self.kernel, [[len(shape) - 1],
|
/external/tensorflow/tensorflow/contrib/model_pruning/python/layers/ |
D | core_layers.py | 458 outputs = standard_ops.tensordot(inputs, self.masked_kernel,
|
/external/tensorflow/tensorflow/contrib/bayesflow/python/ops/ |
D | layers_dense_variational.py | 245 return standard_ops.tensordot(inputs, kernel, axes=[[-1], [0]])
|