Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dimmutable_constant_op.cc61 const Status& allocation_status() const { return allocation_status_; } in allocation_status() function in tensorflow::__anonda76d6cb0111::MemmappedTensorAllocator
93 OP_REQUIRES_OK(ctx, allocator->allocation_status()); in Compute()
Dconv_ops_gpu.h59 Status allocation_status(context_->allocate_temp( in AllocateBytes()
62 if (!allocation_status.ok()) { in AllocateBytes()
Dcudnn_rnn_ops.cc376 Status allocation_status(context_->allocate_temp( in AllocateBytes() local
378 if (!allocation_status.ok()) { in AllocateBytes()
379 return ToExecutorStatus(allocation_status); in AllocateBytes()
423 Status allocation_status(context_->allocate_output( in AllocateBytes() local
425 if (!allocation_status.ok()) { in AllocateBytes()
426 return ToExecutorStatus(allocation_status); in AllocateBytes()
463 Status allocation_status = context_->allocate_persistent( in AllocateBytes() local
465 if (!allocation_status.ok()) { in AllocateBytes()
466 return ToExecutorStatus(allocation_status); in AllocateBytes()
Dfft_ops.cc327 Status allocation_status(context_->allocate_temp( in AllocateBytes() local
330 if (!allocation_status.ok()) { in AllocateBytes()
Dbatch_matmul_op_impl.h269 Status allocation_status(context_->allocate_temp(
271 if (!allocation_status.ok()) {
/external/tensorflow/tensorflow/lite/tools/accuracy/
Drun_tflite_model_op.cc108 auto allocation_status = interpreter_->AllocateTensors(); in Compute() local
109 OP_REQUIRES(context, allocation_status == kTfLiteOk, in Compute()