Searched refs:pruning_mode_t (Results 1 – 1 of 1) sorted by relevance
283 const Tensor* pruning_mode_t; in Compute() local284 OP_REQUIRES_OK(context, context->input("pruning_mode", &pruning_mode_t)); in Compute()286 static_cast<PruningMode>(pruning_mode_t->scalar<int32>()()); in Compute()