Home
last modified time | relevance | path

Searched refs:maybe_output (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmap_kernels.h46 std::unique_ptr<Tensor> maybe_output = ctx->forward_input( in ForwardInputOrCreateNewMap() local
50 if (maybe_output != nullptr && maybe_output->dtype() == DT_VARIANT && in ForwardInputOrCreateNewMap()
51 maybe_output->NumElements() == 1) { in ForwardInputOrCreateNewMap()
52 output_tensor = maybe_output.get(); in ForwardInputOrCreateNewMap()
Ddeserialize_sparse_string_op.cc176 gtl::optional<SparseTensor> maybe_output; in Compute() local
179 maybe_output = SparseTensor::Concat<T>(tensors); \ in Compute()
192 DCHECK(maybe_output); in Compute()
193 SparseTensor& output = maybe_output.value(); in Compute()
Dlist_kernels.cc96 std::unique_ptr<Tensor> maybe_output = c->forward_input( in ForwardInputOrCreateNewList() local
100 if (maybe_output != nullptr && maybe_output->dtype() == DT_VARIANT && in ForwardInputOrCreateNewList()
101 maybe_output->NumElements() == 1) { in ForwardInputOrCreateNewList()
102 output_tensor = maybe_output.get(); in ForwardInputOrCreateNewList()