Home
last modified time | relevance | path

Searched refs:allocation_status (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dimmutable_constant_op.cc63 const Status& allocation_status() const { return allocation_status_; } in allocation_status() function in tensorflow::__anon8058df320111::MemmappedTensorAllocator
104 OP_REQUIRES_OK(ctx, allocator->allocation_status()); in Compute()
Dconv_ops_gpu.h72 Status allocation_status(context_->allocate_temp( in AllocateBytes()
75 if (!allocation_status.ok()) { in AllocateBytes()
Dcudnn_rnn_ops.cc375 Status allocation_status(context_->allocate_temp( in AllocateBytes() local
377 if (!allocation_status.ok()) { in AllocateBytes()
378 return ToExecutorStatus(allocation_status); in AllocateBytes()
421 Status allocation_status(context_->allocate_output( in AllocateBytes() local
423 if (!allocation_status.ok()) { in AllocateBytes()
424 return ToExecutorStatus(allocation_status); in AllocateBytes()
460 Status allocation_status = context_->allocate_persistent( in AllocateBytes() local
462 if (!allocation_status.ok()) { in AllocateBytes()
463 return ToExecutorStatus(allocation_status); in AllocateBytes()
Dfused_batch_norm_op.cc680 Status allocation_status(context_->allocate_temp( in AllocateBytes() local
682 if (!allocation_status.ok()) { in AllocateBytes()
683 return allocation_status; in AllocateBytes()
736 Status allocation_status(context_->allocate_output( in AllocateBytes() local
738 if (!allocation_status.ok()) { in AllocateBytes()
739 return allocation_status; in AllocateBytes()
Dfft_ops.cc376 Status allocation_status(context_->allocate_temp( in AllocateBytes() local
379 if (!allocation_status.ok()) { in AllocateBytes()
Dmatmul_op_impl.h301 Status allocation_status(context_->allocate_temp(
303 if (!allocation_status.ok()) {
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_blas.cc3569 port::Status allocation_status = AllocateWorkspace( in DoBlasInternalImpl() local
3571 if (!allocation_status.ok()) { in DoBlasInternalImpl()
3656 port::Status allocation_status = AllocateWorkspace( in DoBlasInternalImpl() local
3658 if (!allocation_status.ok()) { in DoBlasInternalImpl()