Searched refs:idx_tensor (Results 1 – 5 of 5) 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()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 437 auto idx_tensor = indices.take_front().front().getType().cast<ShapedType>(); in Verify() local
438 Type first_elem_ty = idx_tensor.getElementType(); in Verify()
442 idx_tensor = idx.getType().cast<ShapedType>(); in Verify()
443 elem_ty = idx_tensor.getElementType(); in Verify()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer.cc | 3565 const TensorId idx_tensor = ParseTensorName(reduction_node->input(1)); in TrySimplify() local
3566 if (idx_tensor != TensorId(unique_node->name(), 1)) return Status::OK(); in TrySimplify()
|