Lines Matching refs:checkpoint
40 checkpoint::TensorSliceWriter::CreateBuilderFunction builder_func,
82 checkpoint::TensorSliceWriter writer(filename_t.flat<string>()(0),
90 // checkpoint.
106 OP_REQUIRES_OK(context, checkpoint::ParseShapeAndSlice(
143 checkpoint::TensorSliceReader::OpenTableFunction open_func,
160 std::unique_ptr<checkpoint::TensorSliceReader> allocated_reader;
162 const checkpoint::TensorSliceReader* reader =
166 allocated_reader.reset(new checkpoint::TensorSliceReader(
178 "\" not found in checkpoint files ", file_pattern));
193 OP_REQUIRES_OK(context, checkpoint::ParseShapeAndSlice(
273 checkpoint::ParseShapeAndSlice(shape_and_slice, &parsed_full_shape,
279 " does not match the shape stored in checkpoint: ",
