Home
last modified time | relevance | path

Searched refs:OP_REQUIRES_OK_BOOLEAN (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsparse_conditional_accumulator.h364 OP_REQUIRES_OK_BOOLEAN( in GetAndValidateTensorInputForApplyGrad()
370 OP_REQUIRES_OK_BOOLEAN(ctx, in GetAndValidateTensorInputForApplyGrad()
373 OP_REQUIRES_OK_BOOLEAN(ctx, in GetAndValidateTensorInputForApplyGrad()
377 OP_REQUIRES_OK_BOOLEAN(ctx, in GetAndValidateTensorInputForApplyGrad()
399 OP_REQUIRES_OK_BOOLEAN(ctx, this->ValidateShape(*tensor, has_known_shape)); in GetAndValidateTensorInputForApplyGrad()
425 OP_REQUIRES_OK_BOOLEAN(ctx, ctx->allocate_output(0, {nnz}, &idx_tensor)); in ReturnIdxTensor()
443 OP_REQUIRES_OK_BOOLEAN( in ReturnShapeTensor()
Dconditional_accumulator.h123 OP_REQUIRES_OK_BOOLEAN(ctx, ctx->input("gradient", &grad_tensor)); in GetAndValidateTensorInputForApplyGrad()
125 OP_REQUIRES_OK_BOOLEAN(ctx, this->ValidateShape(*tensor)); in GetAndValidateTensorInputForApplyGrad()
Dconditional_accumulator_base.h173 #define OP_REQUIRES_OK_BOOLEAN(CTX, STATUS) \ macro