Searched refs:idx_tensor (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | listdiff_op_test.py | 40 out_tensor, idx_tensor = array_ops.listdiff(
42 tf_out, tf_idx = self.evaluate([out_tensor, idx_tensor])
46 self.assertEqual(1, idx_tensor.get_shape().ndims)
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | listdiff_op_test.py | 48 out_tensor, idx_tensor = diff_func(x_tensor, y_tensor,
50 tf_out, tf_idx = self.evaluate([out_tensor, idx_tensor])
54 self.assertEqual(1, idx_tensor.get_shape().ndims)
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_conditional_accumulator.h | 423 Tensor* idx_tensor; in ReturnIdxTensor() local
425 OP_REQUIRES_OK_BOOLEAN(ctx, ctx->allocate_output(0, {nnz}, &idx_tensor)); in ReturnIdxTensor()
428 auto idx_tensor_vec = idx_tensor->vec<int64>(); in ReturnIdxTensor()
|