Searched refs:idx_tensor (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | listdiff_op_test.py | 35 out_tensor, idx_tensor = array_ops.listdiff(
37 tf_out, tf_idx = self.evaluate([out_tensor, idx_tensor])
41 self.assertEqual(1, idx_tensor.get_shape().ndims)
|
| /external/tensorflow/tensorflow/python/kernel_tests/data_structures/ |
| D | listdiff_op_test.py | 43 out_tensor, idx_tensor = diff_func(x_tensor, y_tensor,
45 tf_out, tf_idx = self.evaluate([out_tensor, idx_tensor])
49 self.assertEqual(1, idx_tensor.get_shape().ndims)
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_conditional_accumulator.h | 398 Tensor* idx_tensor; in ReturnIdxTensor() local
400 OP_REQUIRES_OK_BOOLEAN(ctx, ctx->allocate_output(0, {nnz}, &idx_tensor)); in ReturnIdxTensor()
403 auto idx_tensor_vec = idx_tensor->vec<int64_t>(); in ReturnIdxTensor()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer.cc | 4149 const TensorId idx_tensor = ParseTensorName(reduction_node->input(1)); in TrySimplify() local
4150 if (idx_tensor != TensorId(unique_node->name(), 1)) return OkStatus(); in TrySimplify()
|