Home
last modified time | relevance | path

Searched refs:output_allocated (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dmap_and_batch_dataset_op.cc259 if (result->output_allocated) { in GetNextInternal()
342 output_allocated(false), in BatchResult()
368 bool output_allocated TF_GUARDED_BY(mu);
502 if (result->output_allocated) { in EnsureOutputAllocated()
521 result->output_allocated = true; in EnsureOutputAllocated()
605 result->output_allocated = reader->Contains( in ReadBatchResult()
632 if (result->output_allocated) { in WriteBatchResult()
/external/tensorflow/tensorflow/core/kernels/
Dfused_batch_norm_op.cc714 if (!output_allocated) { in ~CudnnBatchNormAllocatorInOutput()
729 output_allocated = true; in AllocateBytes()
754 bool output_allocated = false; member in tensorflow::functor::CudnnBatchNormAllocatorInOutput