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.cc267 OP_REQUIRES_OK(context, context->forward_input_or_allocate_temp( in Compute()
Dfractional_max_pool_op.cc251 OP_REQUIRES_OK(context, context->forward_input_or_allocate_temp( in Compute()
Dmaxpooling_op.cc279 OP_REQUIRES_OK(context, context->forward_input_or_allocate_temp( in Compute()
/external/tensorflow/tensorflow/core/util/
Drocm_solvers.cc199 const Status status = context_->forward_input_or_allocate_temp( in forward_input_or_allocate_scoped_tensor()
Dcuda_solvers.cc307 const Status status = context_->forward_input_or_allocate_temp( in forward_input_or_allocate_scoped_tensor()
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.h914 Status forward_input_or_allocate_temp(
919 Status forward_input_or_allocate_temp( in forward_input_or_allocate_temp() function
922 return forward_input_or_allocate_temp(candidate_input_indices, type, shape, in forward_input_or_allocate_temp()
Dop_kernel.cc566 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.cc564 ctx, ctx->forward_input_or_allocate_temp( in Compute()