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

  /external/tensorflow/tensorflow/core/kernels/
dynamic_stitch_op.cc 60 // Check if data0.shape[indices0.dims():] == data1.shape[indices1.dims():]
61 static bool SameExtraShape(const Tensor& data0, const Tensor& indices0,
63 const int extra0 = data0.dims() - indices0.dims();
67 if (data0.dim_size(indices0.dims() + i) !=
103 const Tensor& indices0 = (*indices_inputs)[0]; local
114 c, input_num == 0 || SameExtraShape(data0, indices0, data, indices),
116 "Need data[0].shape[", indices0.dims(), ":] = data[", input_num,
120 ", indices[0].shape = ", indices0.shape().DebugString(),
129 for (int d = indices0.dims(); d < data0.dims(); d++) {
  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
dynamic_stitch_op.cc 185 // Check if data0_shape[indices0.dims():] == data1_shape[indices1.dims():]
187 const TensorShape& indices0,
190 const int extra0 = data0_shape.dims() - indices0.dims();
194 if (data0_shape.dim_size(indices0.dims() + i) !=
  /external/tensorflow/tensorflow/python/kernel_tests/
sparse_ops_test.py 216 indices0 = np.array([0, 13, 10, 33, 32, 14])
220 indices0 = sparse_tensor.SparseTensorValue(
222 np.array(indices0, indices_dtype), np.array(shape, np.int64))
229 return ([sparse_tensor.SparseTensor.from_value(indices0),
    [all...]

Completed in 875 milliseconds