Home
last modified time | relevance | path

Searched refs:forward_input_or_allocate_temp (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dhistogram_op.cc44 TF_RETURN_IF_ERROR(context->forward_input_or_allocate_temp( in Compute()
Dfractional_avg_pool_op.cc322 OP_REQUIRES_OK(context, context->forward_input_or_allocate_temp( in Compute()
Dfractional_max_pool_op.cc274 OP_REQUIRES_OK(context, context->forward_input_or_allocate_temp( in Compute()
Dmaxpooling_op.cc283 OP_REQUIRES_OK(context, context->forward_input_or_allocate_temp( in Compute()
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.h895 Status forward_input_or_allocate_temp(
900 Status forward_input_or_allocate_temp( in forward_input_or_allocate_temp() function
903 return forward_input_or_allocate_temp(candidate_input_indices, type, shape, in forward_input_or_allocate_temp()
Dop_kernel.cc579 Status OpKernelContext::forward_input_or_allocate_temp( in forward_input_or_allocate_temp() function in tensorflow::OpKernelContext
/external/tensorflow/tensorflow/core/kernels/sparse/
Dmat_mul_op.cc573 ctx, ctx->forward_input_or_allocate_temp( in Compute()
/external/tensorflow/tensorflow/core/util/
Drocm_solvers.cc723 const Status status = context_->forward_input_or_allocate_temp( in forward_input_or_allocate_scoped_tensor()
Dcuda_solvers.cc306 const Status status = context_->forward_input_or_allocate_temp( in forward_input_or_allocate_scoped_tensor()