Home
last modified time | relevance | path

Searched refs:forward_input_or_allocate_temp (Results 1 – 7 of 7) 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()
Dcuda_solvers.cc286 const Status status = context_->forward_input_or_allocate_temp( in forward_input_or_allocate_scoped_tensor()
Dmaxpooling_op.cc263 OP_REQUIRES_OK(context, context->forward_input_or_allocate_temp( in Compute()
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.h869 Status forward_input_or_allocate_temp(
874 Status forward_input_or_allocate_temp( in forward_input_or_allocate_temp() function
877 return forward_input_or_allocate_temp(candidate_input_indices, type, shape, in forward_input_or_allocate_temp()
Dop_kernel.cc564 Status OpKernelContext::forward_input_or_allocate_temp( in forward_input_or_allocate_temp() function in tensorflow::OpKernelContext