Searched defs:checkpoint_prefixes (Results 1 – 3 of 3) sorted by relevance
268 const Tensor& checkpoint_prefixes = context->input(0); in Compute() local
416 def get_checkpoint_mtimes(checkpoint_prefixes): argument
490 mlir::Value checkpoint_prefixes = else_fn_block->getArgument(0); in ExpandMergeV2Op() local