HomeSort by relevance Sort by last modified time
    Searched refs:truth_shape (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/python/kernel_tests/
edit_distance_op_test.py 101 truth_shape = [2, 2]
106 truth=(truth_indices, truth_values, truth_shape),
116 truth_shape = [2, 3]
121 truth=(truth_indices, truth_values, truth_shape),
135 truth_shape = [2, 11]
141 truth=(truth_indices, truth_values, truth_shape),
147 truth=(truth_indices, truth_values, truth_shape),
157 truth_shape = [2, 2, 1]
165 truth=(truth_indices, truth_values, truth_shape),
175 truth_shape = [1, 1
    [all...]
reduce_join_op_test.py 102 truth_shape,
111 truth_shape: An array or np.array of the expected shape.
125 self.assertAllEqual(truth_shape, output.get_shape())
176 truth_shape = []
177 self._testReduceJoin(input_array, truth, truth_shape, reduction_indices=0)
212 truth_shape = [2] * 4
215 input_array, truths[i], truth_shape, reduction_indices=i)
220 truth_shape = [2] * 4
223 input_array, truths[i], truth_shape, reduction_indices=i - 5)
231 truth_shape = [2] *
    [all...]
  /external/tensorflow/tensorflow/core/kernels/
edit_distance_op.cc 42 const Tensor& truth_shape) {
63 if (!TensorShapeUtils::IsVector(truth_shape.shape()))
65 "truth_shape should be a vector, but got shape: ",
66 truth_shape.shape().DebugString());
73 if (truth_shape.NumElements() < 2)
75 "Input SparseTensors must have rank at least 2, but truth_shape "
77 truth_shape.NumElements());
78 if (truth_shape.NumElements() != truth_indices.dim_size(1))
80 "Expected truth_shape.NumElements == "
82 truth_shape.shape().DebugString(), " and "
109 const Tensor* truth_shape; variable
    [all...]